ML Compiler Engineer, Compute

Waymo, an autonomous driving technology company, is seeking a talented ML Compiler Engineer to join their Compute Team. Founded as the Google Self-Driving Car Project in 2009, Waymo has established itself as a leader in autonomous driving technology, with their Waymo Driver powering fully autonomous ride-hailing services and accumulating extensive experience across public roads and simulation.

The Compute Team at Waymo plays a crucial role in delivering the compute platform that runs the autonomous vehicle's software stack. This involves creating custom silicon, developing cutting-edge system-level compute architectures, and collaborating across teams to optimize hardware and software performance. As an ML Compiler Engineer, you'll be at the forefront of pushing the boundaries of autonomous vehicle technology.

In this hybrid role, you'll be responsible for maximizing the performance of neural networks through compiler optimization, working closely with hardware architects and model developers to understand and enhance our unique neural network inference platform. Your expertise in compiler development and parallel computing will be essential in implementing support for novel features in our high-performance architecture.

The ideal candidate should have a strong educational background in Computer Science or Electrical Engineering, coupled with practical experience in compilers and ML inference. Proficiency in C++ programming is required, while Python experience and knowledge of neural network architectures would be advantageous. You'll be joining a diverse team of professionals working on one of the world's highest performance automotive compute platforms.

Waymo offers an excellent benefits package including comprehensive health coverage (medical, dental, vision), mental wellness support, gym membership, and financial benefits such as competitive compensation, bonus opportunities, and equity. The company promotes work-life balance through flexible work arrangements, including hybrid work options and generous paid time off policies.

This position presents an exciting opportunity to work at the intersection of autonomous driving technology and high-performance computing, contributing to technology that has the potential to save thousands of lives and transform transportation. You'll be part of a team that values innovation, collaboration, and technical excellence, while working on challenging problems that push the boundaries of what's possible in autonomous vehicle computing.