The opportunity
Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced Driver™—to improve access to…
What you'll do
Conduct comprehensive experimentation to train and deploy state-of-the-art: Multimodal LLMs and World models to perform 3D Perception using sensor information from Camera, LiDAR and Radar..
Partner effectively with engineering and research teams across Waymo to: deploy new models, and implement efficient workflows for model development and continuous training on new front-filled data.
Apply and develop techniques such as quantization, pruning, knowledge: distillation, and efficient attention mechanisms.
Develop and maintain scalable data pipelines for Training & Eval to process data from multiple sources.
Design and implement evaluation frameworks for perception models.
Develop infrastructure for large-scale model distillation and bulk-inference pipelines for teacher models.
What they're looking for
- Experiment with different model partitioning and sharding strategies to improve scalability and efficiency.
- Build and maintain tools for performance analysis, profiling (e.g., xprof), and debugging of ML models.
- PhD or Masters in Computer Science, Machine Learning, Robotics, or a similar: technical field, with 4+ years of industry or post-doc research experience in Reinforcement Learning or Foundation Models.
- Proficiency in implementing model training flows in a scalable, distributed: and performant manner such as Data parallel, FSDP and other sharding approaches.