Waymo
Senior Machine Learning Engineer (Infra), Driver Understanding and Evaluation
Mountain View, CA, USA · Senior · Full-time
Sponsorship not specified$213k-$263kDetected 12 days ago
Distributed SystemsMachine LearningTensorFlowPyTorchData EngineeringRobotics
About the role
- The Waymo Driver powers Waymo's fully autonomous ride-hail service and can also be applied to a range of vehicle platforms and product use cases.
- The Waymo Driver has provided over ten million rider-only trips, enabled by its experience autonomously driving over 100 million miles on public roads and tens of billions in simulation across 15+ U.S. states.
- Translate product/business goals into measurable technical deliverables, ensuring system component alignment.
Responsibilities
- Build scalable systems for training and fine-tuning large-scale models to evaluate interesting driving behaviors.
- Work at the intersection of data engineering, model development, and simulation Provide guidance on architectural decisions and technical directions. Own large, complex systems, driving architectures that meet technical and business objectives.
- Design and scale large distributed systems covering the ML lifecycle, supporting planet-scale dataset generation, model training, and evaluation.
- Collaborate cross-functionally to derive performance and system-level requirements for large ML systems. Translate product/business goals into measurable technical deliverables, ensuring system component alignment.
Requirements
- Strong expertise in distributed training techniques, including gradient sharding and optimization strategies for scaling large models across ML accelerator profiling tools to uncover performance bottlenecks.
Skills
- Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver.
Compensation
- $213,000 - $263,000 USD
Benefits
- 7+ years in machine learning infrastructure such as developing, designing, scaling, training, deploying, and optimizing large-scale machine learning systems from data to model.
- Contribute to the production and optimization of machine learning models aiming to assess Waymo's expansive fleet of vehicles that cumulatively travel millions of miles.
This listing is sourced directly from Waymo's careers page and normalized into a canonical job model.