Machine Learning Engineer II - Autonomous Driving Training Infrastructure

Maymobility; United States; On-site; USD 160,000 - 210,000; Full time

Benefits offered by this job

Comprehensive healthcare suite

Rich retirement benefits

Flexible vacation policy

Total Wellness Program

Job summary

May Mobility in Ann Arbor is seeking ML-Oriented Software Engineers to join their Autonomous Driving ML team. Candidates will transition ML models from concept to commercial scale and work on data infrastructures that support autonomous vehicles. A Bachelor’s or Master’s degree in Robotics, Computer Science, or a related field is required, along with 2+ years of relevant experience. Benefits include comprehensive healthcare and generous paid parental leave.

Salary range is $160,000 - $210,000 USD. Join May Mobility to contribute to innovative transportation solutions!

Qualifications

• Minimum of 2 years building ML-oriented infrastructure in production.
• Proficiency in C++, Python, and PyTorch.
• Familiarity with basic Machine Learning concepts.

Responsibilities

• Transition Autonomous Driving ML models from concept to commercial scale.
• Architect and operate data, training and evaluation pipelines.
• Design and maintain data stores for training workflows.

Skills

C++

Python

PyTorch

Machine Learning concepts

Containerized ML Training workloads

CI/CD pipelines

Education

Bachelor’s or Master’s degree in Robotics, Computer Science or related field

Tools

Linux environments

Ray

Kubeflow

Airflow

MLflow

Weights & Biases

Job description

May Mobility is transforming cities through autonomous technology to create a safer, greener, more accessible world. Based in Ann Arbor, Michigan, May develops and deploys autonomous vehicles (AVs) powered by our innovative Multi‑Policy Decision Making (MPDM) technology that literally reimagines the way AVs think. Our vehicles do more than just drive themselves – they provide value to communities, bridge public transit gaps and move people where they need to go safely, easily and with a lot more fun. We are building the world’s best autonomy system to reimagine transit by minimizing congestion, expanding access and encouraging better land use in order to foster more green, vibrant and livable spaces. Since our founding in 2017, we’ve given more than 500,000 autonomous rides to real people around the globe. And we’re just getting started. We are hiring people who share our passion for building the future, today, solving real‑world problems and seeing the impact of their work. Join us.

Job Summary

May Mobility is entering an exciting phase of growth as we expand our first‑of‑its‑kind autonomous shuttle and mobility services across the nation. Launched in 2017 with a strong team of experienced roboticists and software engineers with decades of experience fielding robotic systems in the wild, May Mobility is looking to expand its team of robotics engineers with a background in robotics or autonomous vehicles. We are seeking ML‑Oriented Software Engineers with experience in robotics applications. As part of our Autonomous Driving ML team, you will use your knowledge of Software and ML concepts to design and operate pipelines that allow May’s Autonomous Driving stack to improve quickly and reliably at scale.

Essential Responsibilities

• Own the process to transition Autonomous Driving ML models from concept to commercial scale.
• Architect and operate data, training and evaluation pipelines across cloud and cluster environments.
• Design and maintain the data and metadata stores that back our training and evaluation workflows.

Skills and Abilities

Success in this role typically requires the following competencies:

• Employ data and model parallelism training techniques for large data (>100TB) or large model (>100GB) applications
• Architecting and operating containerized/pipelined ML Training workloads, including GPU scheduling/autoscaling, dataloader design and experiment tracking.
• Building and maintaining CI/CD pipelines and infrastructure.
• Working with relational and object stores, and high‑throughput data formats for ML workloads.

Qualifications and Experience

Candidates most successful in this role typically hold the following qualifications or comparable knowledge or experience:

Required

• Bachelor’s or Master’s degree in Robotics, Computer Science or a related field with strong mathematical and engineering foundations.
• A minimum of 2 years building ML‑oriented infrastructure, platforms, or distributed systems in production.
• Proficiency in C++, Python and PyTorch with experience in Linux environments.
• Familiarity with basic concepts in Machine Learning (training loops, basic operators and architectures)

Desirable

• Familiarity with basic Perception and Planning concepts in Autonomous Driving.
• Proficiency in Go or Rust.
• Familiarity with ML orchestration and experiment tooling such as Ray, Kubeflow, Airflow, MLflow, or Weights & Biases.
• Familiarity with distributed training frameworks (PyTorch DDP/FSDP, DeepSpeed).
• Familiarity with data pipeline and storage technologies (Spark, Parquet, object storage, feature/metadata stores).

Physical Requirements

• Standard office working conditions which includes but is not limited to:
  • Prolonged sitting
  • Prolonged standing
  • Prolonged computer use

Travel required: Low 5–10%

Benefits and Perks

• Comprehensive healthcare suite including medical, dental, vision, life, and disability plans. Domestic partners who have been residing together at least one year are also eligible to participate.
• Health Savings and Flexible Spending Healthcare and Dependent Care Accounts available.
• Rich retirement benefits, including an immediately vested employer safe harbor match.
• Generous paid parental leave as well as a phased return to work.
• Flexible vacation policy in addition to paid company holidays.
• Total Wellness Program providing numerous resources for overall wellbeing

May Mobility is an equal opportunity employer. All applicants for employment will be considered without regard to race, color, religion, sex, national origin, age, disability, sexual orientation, gender identity or expression, veteran status, genetics or any other legally protected basis. Below, you have the opportunity to share your preferred gender pronouns, gender, ethnicity, and veteran status with May Mobility to help us identify areas of improvement in our hiring and recruitment processes. Completion of these questions is entirely voluntary. Any information you choose to provide will be kept confidential, and will not impact the hiring decision in any way. If you believe that you will need any type of accommodation, please let us know.

Salary Range: $160,000 - $210,000 USD