Senior ML Compiler Engineer, Compute

Waymo, an autonomous driving technology company, is seeking a Senior ML Compiler Engineer to join their Compute Team. This is an exciting opportunity to work on cutting-edge technology that powers self-driving vehicles and has already provided over one million rider-only trips.

The Compute Team plays a crucial role in delivering the compute platform that runs Waymo's autonomous vehicle software stack. The team works on custom silicon development, system-level compute architectures, and optimization of hardware and software for maximum performance. This position offers a unique blend of compiler optimization, machine learning, and hardware/software co-design.

As a Senior ML Compiler Engineer, you'll be working with both hardware architects and model developers to optimize neural network inference performance. Your work will directly impact the efficiency and effectiveness of Waymo's autonomous driving technology. The role requires expertise in compiler development, particularly for parallel architectures, and experience with machine learning inference or linear algebra computation.

The position offers competitive compensation, including a salary range of ₹5,500,000—₹6,650,000 INR, along with comprehensive benefits. These include top-notch medical, dental, and vision insurance, mental wellness support, gym membership, and special wellness programs. Financial benefits include bonus opportunities, equity, and an employees provident fund.

Waymo provides excellent work-life balance with flexible work arrangements, including the ability to work from another location for four weeks per year. The company supports hybrid work models and offers generous paid time off, including parental, bereavement, and sick leave.

This is a hybrid role reporting to a Software Engineering Manager, with some work hours aligning with the Pacific Time zone team. Join Waymo's mission to improve access to mobility while saving thousands of lives now lost to traffic crashes.