Senior/Staff ML Engineer, 3D/4D World Modeling, Simulation
Core
Building state-of-the-art 3D/4D world models and generative systems to create ultra-realistic, controllable simulation environments for testing and validating Waymo's autonomous driving technology.
Role type
Senior/Staff IC Machine Learning Engineer (3D/4D World Modeling & Generative AI)
Builds
Dynamic and semantically rich virtual worlds, 4D world models, and generative systems for autonomous vehicle simulation
Domain
Autonomous driving, 3D computer vision, generative AI
Deliverable
production ML models
Required skills
Large-scale generative model development (Diffusion, Flow Matching), Vision Language Models (VLMs), 3D world modeling, 3D reconstruction and rendering, Python, ML frameworks (Jax/Flax, PyTorch, Tensorflow)
Preferred skills
PhD in relevant field, sensor data simulation (Camera, Lidar, Radar), autonomous systems/robotics experience, large-scale GPU/TPU cluster optimization, C++
Technologies
Diffusion models, Flow Matching, Vision Language Models (VLMs), 3D Gaussian Splatting, Jax/Flax, PyTorch, Tensorflow
Responsibilities
Lead design and deployment of 4D world models and generative systems for sensor/semantics generation; Architect and implement scalable ML pipelines for training and deploying generative models; Build production-ready video generation techniques; Apply VLMs to enhance semantic understanding; Partner with research teams to integrate SOTA research; Mentor and provide technical guidance to engineers
Seniority
Senior/Staff, hands-on IC with mentorship