Taro Logo

ML Compiler Engineer, Compute

Waymo is an autonomous driving technology company developing the Waymo Driver, focused on improving mobility while saving lives through autonomous driving technology.
$170,000 - $216,000
Machine Learning
Mid-Level Software Engineer
Hybrid
1,000 - 5,000 Employees
3+ years of experience
AI · Automotive

Description For ML Compiler Engineer, Compute

Waymo, formerly the Google Self-Driving Car Project, is at the forefront of autonomous driving technology since 2009. As part of their Compute Team, the ML Compiler Engineer role offers an exciting opportunity to work on one of the world's highest performance automotive compute platforms. The position involves optimizing neural network performance through compiler development and enhancement, working directly with hardware architects and model developers to push the boundaries of their unique neural network inference platform.

The role combines deep technical expertise in compiler development with machine learning optimization, requiring strong C++ programming skills and experience with parallel computing. You'll be part of a diverse team responsible for delivering the compute platform that powers Waymo's autonomous vehicle software stack.

This is an excellent opportunity for someone passionate about both systems programming and machine learning, offering the chance to work on cutting-edge technology that directly impacts the future of autonomous driving. The position comes with competitive compensation ($170,000-$216,000), comprehensive benefits, and the flexibility of hybrid work arrangements in either Mountain View, CA or New York City.

The ideal candidate will have at least 3 years of industry experience or equivalent advanced degrees in Computer Science or Electrical Engineering, with specific expertise in compilers and ML inference. You'll be joining a company with a proven track record, having provided over one million rider-only trips and autonomously driven tens of millions of miles across 13+ U.S. states.

Working at Waymo means contributing to a mission-driven organization focused on improving mobility while saving lives through autonomous technology. The company offers excellent career growth opportunities, a collaborative environment, and the chance to work on technology that is actively shaping the future of transportation.

Last updated 2 months ago

Responsibilities For ML Compiler Engineer, Compute

  • Maximize performance of our neural networks by enhancing and extending our production grade compiler
  • Work with hardware architects and model developers to develop understanding of our unique neural network inference platform and neural networks
  • Implement compiler support for novel features of our high-performance architecture

Requirements For ML Compiler Engineer, Compute

Python
  • BS degree in Computer Science/Electrical Engineering or equivalent practical experience and 3+ years of industry experience OR
  • MS degree in Computer Science/Electrical Engineering and 1+ years of industry experience OR
  • PhD Degree in Computer Science/Electrical Engineering or equivalent years of experience
  • 1+ years of industry and/or academic experience with compilers and parallel computing
  • 1+ years of industry and/or academic experience working with ML inference or linear algebra computations
  • C++ programming skills

Benefits For ML Compiler Engineer, Compute

Medical Insurance
Dental Insurance
Vision Insurance
Mental Health Assistance
Equity
Parental Leave
  • Top-notch medical, dental and vision insurance
  • Mental wellness support
  • Gym membership
  • Special wellness programs
  • Competitive compensation
  • Bonus opportunities
  • Equity
  • Employees provident fund
  • Employee discounts
  • Flexibility to work from another location for four weeks per year
  • Paid time off
  • Bereavement leave
  • Sick leave
  • Parental leave

Jobs Related To Waymo ML Compiler Engineer, Compute