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.
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.
Top Skills
What We Do
Waymo is an autonomous driving technology company with a mission to make it safe and easy for people and things to move around. With the Waymo Driver, we can improve the world’s mobility while saving thousands of lives.
Waymo reaches out to candidates from official channels only (e.g. directly from @waymo.com email addresses, or through our recruiters or sourcers who are noted as such on LinkedIn). We do not contact candidates about career opportunities through instant messaging apps like Telegram, email addresses from domains other than waymo.com (such as Gmail addresses), direct messages on Twitter, Facebook, and Instagram, or text messages. Visit waymo.com to check out our official job listings.
