2027 Summer Intern, MS/PhD, Machine Learning Engineer

Waymo is an autonomous driving technology company developing fully autonomous driving systems. The Machine Learning Engineer Intern will train and fine-tune transformer models for driving-log sequences, design offline and end-to-end evaluations, build data and evaluation pipelines, and deliver production-quality code and technical presentations in collaboration with multiple stakeholders.

Responsibilities:

• Train and fine-tune a large multi-task transformer over driving-log sequences, in JAX/Flax on TPUs - iterating on fine-tuning strategies, training data mixtures, and losses to improve evaluation quality for the hillclimbing workflow
• Design and run rigorous offline and end-to-end evaluations - PR-AUC, calibration quality, and metric sensitivity on real hillclimbing A/B runs - and build the dataset and evaluation pipelines needed to produce them
• Land production-quality code in a shared, high-traffic codebase, and communicate results through a design doc, team deep dives, and a final intern presentation, partnering with UEM Core, Data Science, and release-eval stakeholders