Hybrid Summer ML Intern: World Model & Simulation

First Roles; Mountain View (CA); Hybrid; USD 96,000 - 117,000; Full time

Benefits offered by this job

Hybrid onsite internship

Job summary

Waymo is seeking a Summer Intern in Mountain View to develop and evaluate generative models for video and physical world simulation. You will work on 3D/4D representations and cross-sensor data, contributing to the Waymo Driver simulation pipeline.

The role targets MS or PhD candidates in CS/ML/CV/Robotics with strong Python skills and experience in PyTorch or JAX. It is a hybrid onsite internship with competitive hourly pay.

Qualifications

• Pursuing MS or PhD in CS/ML/CV/Robotics.
• Solid background in deep learning and CV with hands-on generative modeling experience.
• Strong coding in Python; practical experience training and debugging models in PyTorch or JAX.

Responsibilities

• Implement, train, and evaluate generative models (diffusion, flow, autoregressive) for video and physical world simulation.
• Experiment with unified 3D/4D and spatiotemporal representations across scenes and sensor modalities.
• Investigate conditioning, fine-tuning, or distillation to improve realism and inference speed for closed-loop simulation.
• Collaborate with mentors across Waymo and frontier AI labs to translate research prototypes into simulation pipelines.

Skills

Python

PyTorch

JAX

Deep learning

Computer vision

Generative modeling

Education

MS or PhD in CS/ML/CV/Robotics

Job description

Waymo is seeking a Summer Intern in Mountain View to develop and evaluate generative models for video and physical world simulation. You will work on 3D/4D representations and cross-sensor data, contributing to the Waymo Driver simulation pipeline.

The role targets MS or PhD candidates in CS/ML/CV/Robotics with strong Python skills and experience in PyTorch or JAX. It is a hybrid onsite internship with competitive hourly pay.