2027 Summer Intern, MS/PhD, AI-driven ML Performance Engineering Intern

Waymo is an autonomous driving technology company building the Waymo Driver and custom machine-learning accelerator technology. The intern will use AI agents to develop, validate, analyze, and optimize low-level kernels for a custom ML accelerator, while creating a harness to automate iterative kernel generation and improvement.

What you'll do:

• Work with AI agents to create high performance low-level kernels in a Domain-Specific language for a custom ML accelerator
• Ensure functional correctness and numerical accuracy of kernels
• Conduct performance analysis of the kernels to identify performance bottlenecks and areas that need improvement
• Develop a custom harness to recursively improve kernel generation and remove the need for human involvement

What they're looking for:

• Currently enrolled in an MS or PhD program in Computer Science, Computer Engineering, Computational Science, or a related quantitative field
• Demonstrated track record of using AI agents and harnesses to improve coding performance
• Working knowledge of key kernels for transformer inference including performance analysis and optimization
• Familiarity with one or more low-level ML accelerator DSLs (CUDA, Triton, Gluon, etc.)
• Solid understanding of modern inference accelerator architectures
• Fluency in Python and/or C++
• Working knowledge of different ML numerical formats and their relative accuracy vs performance tradeoffs

Benefits:

• Hybrid onsite internship position
• Interns are also eligible to participate in the Company’s generous benefits programs, subject to eligibility requirements.