ML Engineer (3D/4D World Modeling, Simulation)

Join Waymo as an ML Engineer, where you will lead the design, development, and deployment of cutting-edge 4D world models and generative systems for ultra-realistic simulation use cases. You will architect and implement scalable ML pipelines, build production-ready video generation techniques, and apply Vision Language Models to enhance semantic understanding. You will also mentor and provide technical guidance to other engineers on the team. Enjoy a comprehensive benefits package, including medical, dental, and vision insurance, mental health support, competitive compensation, and a hybrid work model.

• Lead the design, development, and deployment of cutting-edge 4D world models and generative systems for ultra-realistic simulation.
• Architect and implement scalable and robust ML pipelines for training, evaluating, and deploying large-scale generative models.
• Build and scale production-ready video generation techniques to create dynamic and interactive simulation environments.
• Deep expertise in 3D World Modeling or 3D computer vision
• 5+ years of experience in ML engineering and applied Deep Learning, with a strong portfolio of shipped products or publication record
• Familiarity with 3D reconstruction and rendering techniques (e.g., 3D Gaussian Splatting)
• MS or PhD in Computer Science, Machine Learning, Robotics, or a related field
• Strong programming skills in Python and experience with ML frameworks such as Jax/Flax, PyTorch or Tensorflow
• 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
• PhD and a strong track record of delivering impactful ML products in 3D generative models, world models, or video generation.
• Experience in simulating sensor data (Camera, Lidar, Radar) and/or semantic scenes
• Experience in training and optimizing large scale models on GPU/TPU clusters for efficient serving
• Experience with autonomous systems, robotics, or autonomous vehicle simulation
• Experience in C++ for production systems