Summer ML Intern: Generative Modeling & Simulation

Waymo LLC; Mountain View (CA); Hybrid; USD 80,000 - 113,000; Full time

Job summary

Waymo is seeking an intern in software engineering focused on generative modeling for simulation in a hybrid onsite setting in Mountain View, CA. You will work on training and evaluating diffusion models and related architectures for multi-sensor autonomous driving simulation.

The role targets Masters or PhD students in CS/ML/Robotics with strong Python and deep learning experience, and offers exposure to Waymo's simulation pipeline and collaboration with researchers across Waymo and Google.

Qualifications

• Currently pursuing an MS or PhD in Computer Science, Machine Learning, Computer Vision, Robotics, or a related field.
• Strong background in deep learning and computer vision; hands-on experience with modern generative modeling (e.g., diffusion models, flow matching, VAEs, or transformers).
• Strong coding skills in Python and practical experience training and debugging models in PyTorch or JAX.

Responsibilities

• Implement, train, and evaluate generative models for video and physics-based simulation.
• Experiment with 3D/4D representations that generalize across scenes, platforms, and sensor modalities.
• Investigate conditioning and distillation methods to improve temporal consistency and speed for simulation.

Education

MS or PhD pursuit (CS/ML/Vision/Robotics)

Python

PyTorch

JAX

Tools

PyTorch

JAX

Python

Job description

Waymo is seeking an intern in software engineering focused on generative modeling for simulation in a hybrid onsite setting in Mountain View, CA. You will work on training and evaluating diffusion models and related architectures for multi-sensor autonomous driving simulation.

The role targets Masters or PhD students in CS/ML/Robotics with strong Python and deep learning experience, and offers exposure to Waymo's simulation pipeline and collaboration with researchers across Waymo and Google.