Senior/Staff ML Engineer, 3D/4D World Modeling, Simulation

📋 Description

• Lead the design, development and deployment of 4D world models for ultra-realistic simulation of
• Architect scalable ML pipelines for training, evaluating, and deploying large-scale generative
• Build production-ready video generation techniques (Diffusion, Flow Matching) for dynamic
• Apply Vision Language Models (VLMs) to enhance semantic understanding of world simulations.
• Collaborate with research teams to translate state-of-the-art work into production-ready solutions.
• Mentor and provide technical guidance to other engineers on the team.

🎯 Requirements

• MS or PhD in CS, ML, Robotics, or related field.
• 5+ years in ML engineering and applied deep learning with a track record of shipped products or
• Proven experience with large-scale generative models for video or VLMs.
• Deep expertise in 3D world modeling or 3D computer vision; familiarity with 3D
• Strong Python skills; experience with ML frameworks such as Jax/Flax, PyTorch or TensorFlow.

🎁 Benefits

• Salary range: $213,000 — $263,000 USD.
• Waymo discretionary bonus program, equity incentive plan, and benefits program.