Staff Software Engineer, World Model

About the Role

As a Staff Software Engineer on the World Model team at Waymo, you will be responsible for developing the foundational world model that accurately simulates the physical world. This role involves critical contributions to training, evaluation, and optimization of large-scale models.

Responsibilities

• Develop the foundational world model to accurately simulate the physical world.
• Collaborate with engineering and data teams to tackle key challenges in training the world model on large-scale clusters.
• Develop metrics and evaluation benchmarks to better assess model performance.
• Design and implement a scalable and efficient data annotation pipeline to ensure high-quality labeled data for training and evaluation.
• Optimize inference efficiency to enable real-time interaction.