Machine Learning Engineer (GPU Kernel and Runtime)

TL;DR

Machine Learning Engineer (GPU Kernel and Runtime): Developing and optimizing GPU kernels and runtime systems for autonomous driving with an accent on high-performance computing and hardware acceleration. Focus on maximizing throughput and minimizing latency for ML models deployed on vehicle hardware.

Location: Onsite in Mountain View, California

Salary: $213,000—$263,000 USD

Company

Waymo is an autonomous driving technology company building the World's Most Experienced Driver to improve mobility and save lives.

What you will do

• Develop and optimize high-performance GPU kernels for machine learning workloads.
• Design and implement efficient runtime systems for autonomous driving platforms.
• Collaborate with software and hardware teams to optimize AI model execution.
• Improve the scalability and performance of the Waymo Driver's computational stack.

Requirements

• Strong experience in GPU programming and kernel development.
• Proficiency in ML runtime optimization and hardware acceleration.
• Must be based in or be able to work onsite in Mountain View, California.

Culture & Benefits

• Eligibility for a discretionary annual bonus program.
• Equity incentive plan.
• Generous comprehensive company benefits program.