ML Engineer, Foundation Model Infrastructure
WaymoAutonomous Vehicles company
Mountain View, United States$175,000 - $215,000 USDSenior
Dragoneer Investment Group
DST Global
Sequoia Capital
Andreessen Horowitz
Alphabet
Mubadala Capital
Data & AI
About the role
TL;DR
ML Engineer to build and operate ML infrastructure for foundation models.
- •Waymo is seeking an ML Engineer to build and operate petabyte-scale data systems and ML pipelines for foundation model development in autonomous driving.
- •You will shepherd cutting-edge models from research to production.
- •Key Responsibilities Build and operate petabyte-scale data systems and ML pipelines for foundation model development.
- •Shepherd foundation models from research prototypes to robust components within the Waymo Driver.
- •Create automated infrastructure for benchmarking, monitoring, and releasing models.
- •Utilize large-scale compute and frameworks like Flume and JAX.
- •Drive improvements in the speed, reliability, and efficiency of the ML development lifecycle.
- •Requirements Master's degree in Computer Science, Machine Learning, Robotics, or equivalent practical experience.
- •Proficiency in Python and C++.
- •Familiarity with 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).
Required skills
PythonC++JAXSparkKubeflow
Domain expertise
automotivedeeptechai
Tech stack
PythonC++JAXSparkKubeflow