Foundation ML Engineer: Scale Data & Models for Driving

Waymo; Mountain View (CA); Hybrid; USD 175,000 - 215,000; Full time

Benefits offered by this job

Health, dental, vision, life, and more

401(k) with company match

Paid Time Off: 20 days/year

Sick Time: 40 hours/year

Maternity Leave: 28-30 weeks

Baby Bonding Leave: 18 weeks

13 paid holidays

Job summary

Waymo is seeking a highly skilled ML Engineer to join the AI Foundations team in a hybrid role reporting to a Senior Research Scientist. You will build petabyte-scale data systems and ML pipelines, shepherd foundation models from research prototypes to robust on-road components, and create automated benchmarking and deployment infrastructure.

Qualifications include a Masters in CS/ML/Robotics or equivalent, proficiency in Python and C++, and experience with PyTorch or JAX, plus large-scale data

Qualifications

• Masters degree in Computer Science, Machine Learning, Robotics, or equivalent practical experience.
• Proficiency in Python and 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).

Responsibilities

• Build and operate the petabyte-scale data systems and ML pipelines at the heart of Waymo's foundation model development
• Shepherd cutting-edge foundation models from research prototypes to robust components within the Waymo Driver
• Create automated infrastructure for benchmarking, monitoring, and safely releasing models
• Wield large-scale compute and frameworks to process massive datasets and train/deploy complex models
• Drive leaps in speed, reliability, and efficiency of the end-to-end ML development lifecycle
• Partner with AI Foundations, ML, and Platform experts to translate model innovations into on-road improvements

Skills

Python

C++

PyTorch

JAX

TensorFlow

Education

Master's degree in CS/ML/Robotics or equivalent

Tools

Flume

Spark

Borg

Kubeflow

Job description

Waymo is seeking a highly skilled ML Engineer to join the AI Foundations team in a hybrid role reporting to a Senior Research Scientist. You will build petabyte-scale data systems and ML pipelines, shepherd foundation models from research prototypes to robust on-road components, and create automated benchmarking and deployment infrastructure.

Qualifications include a Masters in CS/ML/Robotics or equivalent, proficiency in Python and C++, and experience with PyTorch or JAX, plus large-scale data