ML Engineer, Foundation Model Infrastructure
$175,000–$215,000 year
HybridMountain View Santa Clara County, California, United States
Job Summary
Build and operate petabyte-scale data systems and ML pipelines central to Waymo's foundation model development. Shepherd cutting-edge models from research prototypes to robust components within the Waymo Driver, creating automated infrastructure for rigorous benchmarking, continuous monitoring, and safe model releases. Wield large-scale compute and frameworks like Flume and JAX to train and deploy complex models, driving significant leaps in the speed, reliability, and efficiency of the end-to-end ML development lifecycle. Partner with AI Foundations, ML, and Platform experts to transform model innovations into tangible on-road improvements.
Required Qualifications
- Masters degree in Computer Science, Machine Learning, Robotics, similar technical field of study, or equivalent practical experience
- Proficiency in Python
- Proficiency in C++
- Familiarity with one of the modern deep learning frameworks (e.g. Pytorch, JAX, Tensorflow)
- Experience building or maintaining large-scale data pipelines or ML infrastructure (e.g., Flume, Spark, Borg, Kubeflow)
Desired Qualifications
- Strong hands-on SWE skills, able to drive development of large, complex shared codebases
- Experience in AV planning and related research
- Experience designing and building distributed systems or MLOps platforms (e.g., model versioning, experiment tracking, CI/CD for ML)
- Prior work in an industrial or research setting developing methodologies for the evaluation of ML models
Hiring someone like this?
Get your role in front of qualified candidates on Sorce.