Senior Machine Learning Engineer (Simulation Evaluation)

Join Waymo, a leader in autonomous driving technology, as a Senior Machine Learning Engineer. In this role, you will work on one of the most complex virtual environments ever built, the Waymo simulator. You will lead the design, development, and deployment of cutting-edge evaluation approaches to assess the realism of multimodal world models and generative systems for simulation use cases. You will also architect and implement robust and scalable machine learning pipelines, evaluate video generation techniques, and apply vision language models to evaluate semantic understanding and controllability. Additionally, you will collaborate with research teams across Waymo and Alphabet and mentor engineers on the team.

• Lead the design, development, and deployment of cutting-edge evaluation approaches to assess the realism of state-of-the-art multimodal world models and generative systems for simulation use cases at Waymo.
• Architect and implement robust and scalable machine learning pipelines for tuning, evaluating, and deploying large-scale discriminator models for the purposes of simulator realism evaluation.
• Collaborate with research teams across Waymo and Alphabet to integrate advancements in 4D world modeling and generative AI into production systems.
• Proficient programming skills in Python and hands-on experience with modern machine learning frameworks such as Jax, Flax, or PyTorch
• Five or more years of experience in machine learning engineering or applied deep learning, supported by a portfolio of shipped products or peer-reviewed publications
• Experience designing and implementing evaluation frameworks for complex systems or machine learning models
• Bachelor's, Master's, or PhD in computer science, machine learning, robotics, or a related field
• Track record of training large-scale generative models (diffusion models, flow matching, vision language models, etc.)
• A PhD and demonstrated success delivering machine learning products focused on 3D generative models, world models, or video generation
• Experience simulating sensor data, including camera, lidar, and radar, or modeling semantic scenes
• Experience developing autonomous systems, robotics software, or autonomous vehicle simulations
• Experience training and optimizing large-scale models on GPU or TPU clusters for efficient production serving
• Professional experience writing C++ for high-performance production systems