Waymo is looking to solve the problem of improving access to mobility while saving thousands of lives lost to traffic crashes by building the Waymo Driver—The World's Most Experienced Driver™, and the Simulation ML Infrastructure team is focused on building scalable AI/ML infrastructure to accelerate the Simulator team in sustainably innovating and building state of the art simulations of realistic environments for the testing and training of the Waymo Driver.
Requirements
- BS in Computer Science, Robotics, similar technical field of study, or equivalent practical experience
- 5+ years of professional software engineering experience, with at least 3 years in machine learning infrastructure
- Solid experience in the development and optimization of machine learning infrastructure tools like DeepSpeed, PyTorch, TensorFlow, or similar frameworks.
- 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 auto-regressive transformers
- Familiarity with custom-kernels for diverse h/w compute based efficiency
- Practical familiarity in Autonomous Driving, Simulations, and ML accelerators
Responsibilities
- Be part of a world-class, high-performing research engineering team to advance the state of the art of ultra realistic multi-agent simulations using foundation models.
- Collaborate closely with the core Google DeepMind and Waymo Realism Modeling teams in London, and Waymo Oxford to use the large models to improve sim realism.
- Provide deep technical leadership on large-scale ML model architectures, especially for autonomous vehicle models.
- Design and scale large distributed systems covering the ML lifecycle, supporting planet-scale dataset generation and model training.
- Collaborate cross-functionally to derive performance and system-level requirements for large ML systems.
- Translate product/business goals into measurable technical deliverables, ensuring system component alignment.
- Mentor junior engineers, growing their expertise and fostering a collaborative culture.
Other
- BS in Computer Science, Robotics, similar technical field of study, or equivalent practical experience
- 5+ years of professional software engineering experience
- Excellent communication skills, both verbal and written, with the ability to translate complex technical concepts for a broad audience
- Ability to work in a collaborative culture
- Eligibility to participate in Waymo’s discretionary annual bonus program, equity incentive plan, and generous Company benefits program
