Machine Learning Engineer II

About the role

• About Our Client:

The organization operates in the autonomous vehicle industry, focusing on transforming urban transportation to create safer, greener, and more accessible cities. It addresses challenges in public transit by deploying autonomous shuttles powered by advanced Multi-Policy Decision Making technology. Since 2017, the program has provided over 500,000 autonomous rides, aiming to reduce congestion, expand transit access, and promote better urban land use.

• About the Opportunity:

The Machine Learning Engineer II plays a pivotal role in scaling the autonomous driving technology by designing and managing machine learning pipelines and infrastructure. This position supports the organization’s goal to improve its autonomous vehicle systems efficiently and reliably, contributing directly to the expansion and operational success of its mobility services.

• Responsibilities:

Manage the transition of autonomous driving ML models from concept to commercial deployment.

Architect and maintain training and evaluation pipelines in cloud and cluster environments.

Design and oversee data and metadata storage systems supporting ML workflows.

• Requirements:

Bachelor’s or Master’s degree in Robotics, Computer Science, or related field with strong math and engineering foundations.

Minimum of 2 years experience building ML-oriented infrastructure or distributed systems in production.

Proficiency in C++, Python, and PyTorch within Linux environments.

Understanding of machine learning fundamentals such as training loops and model architectures.

Experience with data/model parallelism for large-scale ML applications.

Skill in containerized ML workload orchestration, including GPU scheduling and autoscaling.

Experience with CI/CD pipelines and data storage technologies relevant to ML.

Desirable:

Knowledge of perception and planning in autonomous driving.

Experience with Go or Rust programming languages.

Familiarity with ML orchestration tools (Ray, Kubeflow, Airflow, MLflow, Weights & Biases).

Experience with distributed training frameworks (PyTorch DDP/FSDP, DeepSpeed).

Knowledge of data pipeline and storage tools (Spark, Parquet, object storage).

• Pay Range and Compensation Package:

Salary range: $160,000—$210,000 USD

• Benefits & Perks:

Comprehensive healthcare including medical, dental, vision, life, and disability insurance.

Health Savings and Flexible Spending Accounts for healthcare and dependent care.

Retirement benefits with immediate employer safe harbor match.

Paid parental leave with phased return to work.

Flexible vacation policy and paid company holidays.

Wellness program resources supporting overall employee wellbeing.

Equal Opportunity Statement: Our client is an equal opportunity employer. They celebrate diversity and are committed to creating an inclusive environment for all employees. All qualified applicants will receive consideration for employment without regard to race, color, religion, gender, gender identity or expression, sexual orientation, or national origin.

Note:

RemoteHunter is a recruitment partner of this role. Please note that all employment decisions, including candidate assessment, interviews, hiring, compensation, and employment terms, are made exclusively by the hiring employer.