2027 Summer Intern, PhD, Planner Machine Learning

Tech Stack

Solid badges = required, outlined = preferred

Responsibilities

• Research, design, and implement novel deep learning architecture, multi-task loss formulations, and preference/ranking objectives in Python and JAX
• Train and diagnose large-scale neural networks using distributed TPU infrastructure, analyzing training dynamics, gradient conditioning, and representation quality
• Evaluate trained models across open-loop and closed-loop benchmarks, triaging real-world driving scenarios to connect mathematical modeling choices with physical vehicle behavior

Soft Skills

Benefits

• Hybrid Work

Culture

Requirements