2027 Summer Intern, PhD, Machine Learning - Simulation & Closed-Loop Training

Tech Stack

Responsibilities

• Design, implement, and scale Reinforcement Learning post-training and distillation algorithms for foundation world models in simulation and autonomous driving.
• Develop learned reward models, realism critics, and discriminators within a closed-loop multi-agent simulator.
• Run rigorous closed-loop ablation studies on large-scale TPU clusters using fleet-scale driving behavior metrics.
• Write clean, modular, and well-tested JAX code in core simulation and autonomous driving codebases.

Benefits

• Equity
• Gym Membership

Culture

Requirements