Senior/Staff Machine Learning Engineer

<div class="content-intro"><p>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 mobility while saving thousands of lives now lost to traffic crashes. The Waymo Driver powers Waymo’s fully autonomous ride-hail service and can also be applied to a range of vehicle platforms and product use cases. The Waymo Driver has provided over ten million rider-only trips, enabled by its experience autonomously driving over 100 million miles on public roads and tens of billions in simulation across 15+ U.S. states.</p></div><p>The Simulator Team at Waymo builds state-of-the-art simulations of realistic environments for testing, training, and validation of the Waymo Driver. Our team is a diverse, and collaborative group of machine learning (ML) engineers, software engineers, and ML research engineers. We develop industry-leading simulation solutions using advanced generative and reconstructive ML algorithms, to model the real world, encompassing realistic agents, roads, traffic systems, weather, and the full sensor suite (Camera, Lidar, Radar).</p> <p>To accelerate the fidelity, scalability, controllability, and richness of our simulations, we are pushing the frontiers of 3D world modeling. We leverage state-of-the-art ML technologies trained on large-scale datasets to create dynamic and semantically rich virtual worlds, directly impacting the development and validation of the Waymo Driver.</p> <p>In this role, you will report to a Senior Staff Engineering Manager.</p> <p> </p> <p><strong>You will:</strong></p> <ul> <li>Lead the design, development and deployment of cutting-edge 4D world models and generative systems for ultra-realistic and controllable sensor and semantics generation for simulation use cases at waymo.</li> <li>Architect and implement scalable and robust ML pipelines for training, evaluating, and deploying large-scale generative models into our simulation infrastructure, including techniques like model distillation and quantization.</li> <li>Build and scale production-ready video generation techniques (e.g., Diffusion, Flow Matching) to create dynamic and interactive simulation environments.</li> <li>Apply Vision Language Models (VLMs) to enhance the semantic understanding and controllability of our world simulation products.</li> <li>Partner with world class research teams across Waymo and Alphabet to leverage State-of-The-Art research in 4D world modeling and generative AI into robust, production-ready solutions.</li> <li>Mentor and provide technical guidance to other engineers on the team.</li> </ul> <p> </p> <p><strong>You have:</strong></p> <ul> <li>MS or PhD in Computer Science, Machine Learning, Robotics, or a related field.</li> <li>5+ years of experience in ML engineering and applied Deep Learning, with a strong portfolio of shipped products or publication record.</li> <li>Proven experience in developing and training large-scale generative models for video generation (e.g., Diffusion models, Flow Matching) or Vision Language Models (VLMs) and their applications.</li> <li>Deep expertise in 3D World Modeling or 3D computer vision.</li> <li>Familiarity with 3D reconstruction and rendering techniques (e.g., 3D Gaussian Splatting).</li> <li>Strong programming skills in Python and experience with ML frameworks such as Jax/Flax, PyTorch or Tensorflow.</li> </ul> <p> </p> <p><strong>We prefer:</strong></p> <ul> <li>PhD and a strong track record of delivering impactful ML products in 3D generative models, world models, or video generation..</li> <li>Experience in simulating sensor data (Camera, Lidar, Radar) and/or semantic scenes.</li> <li>Experience with autonomous systems, robotics, or autonomous vehicle simulation.</li> <li>Experience in training and optimizing large scale models on GPU/TPU clusters f ... (truncated, view full listing at source)