2026 Summer Intern, PhD, ML Compilers

The ML Runtime and Deployment Team supports and facilitates the fast deployment and execution of neural nets on specialized hardware used in the car and at scale for simulation in data centers. The team builds the infrastructure for easy model launches and optimizes the performance of the onboard software through system-wide performance tunings, model-level optimization as well as hardware-software co-play on specialized accelerators.

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:

• Develop innovative compiler optimization techniques to enhance the efficiency of Waymo ML models.
• Prototype novel programming interfaces for authoring highly efficient and adaptable ML algorithms and kernels.
• Implement solutions using C++ and Python.

You have:

• Progressing towards PhD in Computer Science, Electrical Engineering, Machine Learning, or related technical fields.
• Experience in ML compilers and MLIR.
• Strong programming skills in C++.

We prefer:

• Strong track record of high quality research in machine learning, computer systems, and/or computer architecture.
• Experience with hardware acceleration, GPU and/or TPU optimizations.
• Experience with JAX, Pallas and Triton.

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.