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

In this role, you will report to a Senior Staff Engineering Manager 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 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 Build and scale production-ready video generation techniques (e.g., Diffusion, Flow Matching) to create dynamic and interactive simulation environments Apply Vision Language Models (VLMs) to enhance the semantic understanding and controllability of our world simulation products 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 Mentor and provide technical guidance to other engineers on the team Benefits Medical, dental, and vision insurance for employees and dependents Employee assistance programs focused on mental health Personalized workplace adjustments for diverse needs and abilities, including physical, mental, and neurodivergent considerations Access to mental health apps Onsite wellness centers Counseling services Second medical opinion for you and your loved ones Medical advocacy program for transgender employees Support programs including menopause benefit Competitive compensation Regular bonus and equity performance grant opportunities Generous 401(k) and regional retirement plans Annual cross-company compensation review and pay equity analysis 1-on-1 financial coaching Fertility and growing family assistance Parental leave and baby bonding leave Survivor income benefit Elder care and support Backup childcare Caregiver leave Paid time off, including vacation, bereavement, sick leave, parental leave, disability, and holidays Jury duty leave Military leave Hybrid work model with remote work opportunities also available Educational reimbursement Peer learning and coaching platform Donation matching programs Employee resource groups Volunteer hours Internal community groups and local culture clubs Inspiring spaces to work, recharge, and collaborate with fellow Waymonauts On-site meals and snacks Fitness centers, massage programs, and ergonomic support On-demand fitness, wellbeing, and cooking classes Commuter benefits Qualifications 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 #J-18808-Ljbffr Waymo