Waymo is looking to improve the performance of its autonomous driving technology by developing machine learning techniques for evaluation systems on its autonomous vehicles
Requirements
- M.S. or Ph.D. degree Computer Science, Machine Learning, Artificial Intelligence, or a related technical field, or equivalent practical experience
- 5+ years of professional software engineering experience, with at least 3 years in machine learning infrastructure
- A history of contributions to machine learning tooling and frameworks e.g. PyTorch, Jax, Tensorflow, Ray, or similar
- Strong expertise in distributed training techniques, including gradient sharding and optimization strategies for scaling large models
- Deep understanding of state-of-the-art machine learning models such as autoregressive transformers
- Experience in the autonomous vehicles domain, robotics, or complex simulation environments
- Deep understanding of state-of-the-art RL techniques, including those used for fine-tuning large models
Responsibilities
- Provide deep technical leadership on large-scale ML model architectures, especially for autonomous vehicle models
- 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
- Oversee the production and optimization of machine learning models aiming to assess Waymo’s expansive fleet of vehicles that cumulatively travel millions of miles
- 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
Other
- M.S. or Ph.D. degree
- 5+ years of professional software engineering experience
- Strong leadership skills with experience navigating cross-functional teams and providing technical leadership projects across multiple organizations
- Excellent communication skills, with the ability to articulate complex technical concepts clearly
- Track record of technical leadership, influencing senior stakeholders, and driving innovation across team boundaries
