2027 Summer Intern MS PhD Machine Learning Engineer

Waymo is an Alphabet company operating a public, fully autonomous ride-hailing service and developing the AI-powered Waymo Driver autonomous-driving system.

0 current maintainers0 active leads
Mountain View, United States
About Waymo

Waymo develops and deploys the Waymo Driver, an autonomous-driving technology using AI and a sensor suite, and operates its ride-hailing service for public riders.

View jobs by Waymo

Skills

Candidate Availability

Required and preferred rules are kept separate and reflect the wording in the original posting.

About the Role

You will train and fine-tune a multi-task transformer using JAX and Flax on TPUs, design offline and end-to-end evaluations, build data and evaluation pipelines, deliver production-quality code, and communicate results through technical documentation and presentations.

Requirements

  • Enrollment in a PhD or MS program in Computer Science, Machine Learning, or a related field
  • Deep learning model training and evaluation
  • JAX, PyTorch, or TensorFlow
  • Data pipeline development
  • C++
  • Python
  • Machine learning fundamentals
  • Experiment design

Responsibilities

  • Train and fine-tune a multi-task transformer over driving-log sequences
  • Design and run offline and end-to-end evaluations
  • Build datasets and evaluation pipelines
  • Deliver production-quality code in a shared codebase
  • Communicate results through a design document, deep dives, and a final presentation

Benefits

  • Housing or relocation bonus if applicable
  • Medical, dental, and vision insurance
  • Intern events and networking opportunities
  • Free breakfast, lunch, dinner, and snacks
  • Free access to Google shuttles
  • Onsite gym
  • Mental wellness support
  • Gym membership
  • Wellness programs
  • Bonus opportunities
  • Equity
  • Employee discounts
  • Paid time off
  • Bereavement leave
  • Sick leave
  • Parental leave
  • Education reimbursement
  • Employee Resource Groups
  • Volunteer time off
2027 Summer Intern MS PhD Machine Learning Engineer at Waymo | JobStash