About the Role
Waymo is seeking an individual to join the AI Foundations team, focusing on developing machine learning solutions for autonomous driving. This role involves building and operating large-scale data systems and ML pipelines to advance foundation model development for the Waymo Driver.
Responsibilities
- Build and operate petabyte-scale data systems and ML pipelines for foundation model development.
- Shepherd cutting-edge foundation models from research prototypes to robust components within the Waymo Driver.
- Develop automated, high-performance infrastructure for efficient rollout of data solutions.
- Process massive datasets using large-scale compute and frameworks like Flume/Beam and JAX for model training, evaluation, and deployment.
- Drive significant improvements in the speed, reliability, and efficiency of the end-to-end ML development lifecycle.
- Partner with experts in AI Foundations, ML, and Platform to translate model innovations into product improvements.
Requirements
- Undergraduate degree in Computer Science, Machine Learning, Robotics, a similar technical field, or equivalent practical experience.
- Proficiency in Python and C++.
- Experience building or maintaining large-scale data pipelines or ML infrastructure (e.g., Flume, Spark, Borg, Kubeflow).
- Strong hands-on SWE skills, capable of driving development of large, complex shared codebases.
Skills
- Python
- C++
- Flume
- Beam
- JAX
- Spark
- Borg
- Kubeflow
- Pytorch
- Tensorflow
- Distributed systems
- MLOps platforms
- Model versioning
- Experiment tracking
- CI/CD for ML
- AV planning
- Data-centric improvement of ML models
- Data curation
- Multimodal experience
Location
- US
Work Type
- Hybrid
- Full-time
Experience Level
- Mid-level
- Senior
Education Level
- Undergraduate degree
- Computer Science
- Machine Learning
- Robotics
- Technical field
- Equivalent practical experience
Salary/Compensations
- $213,000—$263,000 USD
Benefits
- Health, dental, vision, life, disability insurance
- 401(k) with company match
- 20 days of vacation per year
- 40 hours/year sick time
- 5 days/event sick time
- 28-30 weeks maternity leave
- 18 weeks baby bonding leave
- 13 paid holidays per year
- Discretionary annual bonus program
- Equity incentive plan
About the Company
- Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver.
- Started as the Google Self-Driving Car Project in 2009.
- Focused on building the Waymo Driver—The World's Most Experienced Driver™.
- Aims to improve mobility access and reduce traffic crash fatalities.
- The Waymo Driver powers Waymo’s fully autonomous ride-hail service and can be applied to various vehicle platforms and product use cases.
- Has provided over ten million rider-only trips.
- Driven over 100 million miles autonomously on public roads and tens of billions in simulation across 15+ U.S. states.
Equal Opportunity
- In accordance with Washington state law, we are highlighting our comprehensive benefits package, which is available to all eligible US based employees.
