Senior Machine Learning Engineer (Infra), Driver Understanding and Evaluation

Location: On site in Mountain View, California, United States

Salary: $213,000–$263,000 USD annual base salary, plus potential annual bonus, equity, and company benefits.

Company

Waymo develops autonomous driving technology and fully autonomous ride-hailing services through the Waymo Driver.

What you will do

• Build scalable systems for training and fine-tuning large-scale models that evaluate driving behaviors.
• Develop and operate machine learning infrastructure, data systems, simulation workflows, and evaluation tools.
• Contribute to the production and optimization of models assessing Waymo's autonomous vehicle fleet.
• Design distributed systems for dataset generation, model training, and evaluation across the ML lifecycle.
• Provide architectural and technical direction for complex systems and collaborate cross-functionally on measurable deliverables.

Requirements

• Master's or Ph.D. in Computer Science, Machine Learning, Artificial Intelligence, or a related technical field, or equivalent practical experience.
• 5+ years of experience developing, scaling, training, deploying, and optimizing large-scale machine learning systems.
• Experience with machine learning tooling and frameworks such as PyTorch, Jax, TensorFlow, Ray, or similar.
• Strong expertise in distributed training, gradient sharding, model-scaling optimization, and ML accelerator profiling.
• Ability to work on site in Mountain View, California.

Nice to have

• 7+ years of experience in machine learning infrastructure.
• Experience in autonomous vehicles, robotics, or complex simulation environments.
• Familiarity with large-scale simulation platforms and their integration with ML training workflows.

Culture & Benefits

• Eligibility for a discretionary annual bonus program.
• Equity incentive plan and company benefits program, subject to eligibility.
• Work on autonomous driving systems operating across a large vehicle fleet and extensive simulation infrastructure.