2026 Summer Intern, MS/PhD, ML Architecture, Software Engineer

Descrizione del lavoro:

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver-The World's Most Experienced Driver™-to improve access to mobility while saving thousands of lives now lost to traffic crashes. The Waymo Driver powers Waymo's fully autonomous ride-hail service and can also be applied to a range of vehicle platforms and product use cases. The Waymo Driver has provided over ten million rider-only trips, enabled by its experience autonomously driving over 100 million miles on public roads and tens of billions in simulation across 15+ U.S. states.
Waymo interns partner with leaders in the industry on projects that create impact to the company. We believe learning is a two-way street: applying your knowledge while providing you with opportunities to expand your skill-set. Interns are an important part of our culture and our recruiting pipeline. Join us at Waymo for a fun and rewarding internship!
You will:
* Understand difficult-to-visualize manipulations and computations on 4-5 dimensional tensors, devise ways to express them cleanly and implement them on domain-specific architectures
* Deep dive into the capabilities of unfamiliar hardware to map operations to it
* Quickly create minimally featured software prototypes with light automated testing to explore a design space
You have:
* GPU, Pallas or other accelerator programming
* Computer architecture fundamentals
* Professional competency in C++ or Python
* Currently pursuing a PhD in related field or Masters with significant industry experience
We prefer:
* Experience with compilers for CPU or GPU or ML accelerators
* Experience with NVIDIA CUTLASS, CuTile, Pallas, or Triton-like Domain Specific Languages
* Experience with MLIR, JAX, ONNX, StableHLO or similar compiler intermediate representations
Note: This will be a hybrid onsite internship position. We will accept resumes on a rolling basis until the role is filled. To be in consideration for multiple roles, you will need to apply to each one individually - please apply to the top 3 roles you are interested in.

The expected hourly rate for this full-time position is listed below. Interns are also eligible to participate in the Company's generous benefits programs, subject to eligibility requirements.
Hourly Masters Pay
$70—$70 USD

The expected hourly rate for this full-time position is listed below. Interns are also eligible to participate in the Company's generous benefits programs, subject to eligibility requirements.
Hourly PhD Pay
$85—$85 USD