Senior Machine Learning Engineer (Simulation Evaluation)

TL;DR

Senior Machine Learning Engineer (Generative AI/Simulation): Architecting scalable deep learning systems and novel data workflows to evaluate the realism of multimodal world models with an accent on generative vision paradigms. Focus on designing discriminator models, implementing evaluation frameworks for 4D world modeling, and integrating generative AI into production autonomous driving systems.

Location: Onsite in Mountain View, California

Salary: $213,000—$263,000 USD

Company

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver.

What you will do

• Lead the design and deployment of cutting-edge evaluation approaches to assess realism of multimodal world models and generative systems.
• Architect and implement scalable ML pipelines for tuning, evaluating, and deploying large-scale discriminator models.
• Evaluate open-source and production-ready video generation techniques focusing on temporal stability and multi-modal consistency.
• Apply vision language models to evaluate semantic understanding and controllability across world simulation products.
• Collaborate with research teams across Waymo and Alphabet to integrate advancements in 4D world modeling and generative AI.
• Mentor engineers on the team and provide technical guidance on architecture and execution.

Requirements

• Bachelor's, Master's, or PhD in computer science, machine learning, robotics, or a related field.
• Five or more years of experience in machine learning engineering or applied deep learning.
• Proficient programming skills in Python and experience with Jax, Flax, or PyTorch.
• Experience designing and implementing evaluation frameworks for complex systems or ML models.
• Must be based in or work onsite in Mountain View, California.

Nice to have

• Track record of training large-scale generative models (diffusion models, flow matching, VLMs).
• PhD with success delivering ML products focused on 3D generative models or video generation.
• Experience simulating sensor data (camera, lidar, radar) or modeling semantic scenes.
• Experience developing autonomous systems, robotics software, or AV simulations.
• Experience optimizing large-scale models on GPU or TPU clusters.
• Professional experience writing C++ for high-performance production systems.

Culture & Benefits

• Discretionary annual bonus program.
• Equity incentive plan.
• Generous Company benefits program.
• Opportunity to work on one of the most complex virtual environments ever built using state-of-the-art Generative AI.