2027 Summer Intern, MS/PhD, Machine Learning Engineer
Core
Train and fine-tune large multi-task transformers over driving-log sequences to improve evaluation quality for autonomous driving workflows.
Role type
Summer Intern, Machine Learning Engineer (Research)
Builds
Evaluation pipelines and datasets for the Waymo Driver's hillclimbing workflow
Domain
Autonomous driving, deep learning, robotics
Deliverable
production ML models
Required skills
deep learning model training and evaluation, transformer and sequence models, multi-task learning, JAX/Flax, C++, Python, experiment design
Preferred skills
parameter-efficient fine-tuning, distributed training on TPUs, probability calibration, uncertainty quantification, autonomous driving simulation
Technologies
JAX, Flax, TPUs, Python, C++, SQL
Responsibilities
Iterate on fine-tuning strategies and training data mixtures; design and run rigorous offline and end-to-end evaluations; land production-quality code and communicate results via design docs and presentations
Seniority
Intern