ML Compiler Engineer, Compute

Waymo's Compute Team is tasked with a critical and exciting mission: We deliver the compute platform responsible for running the autonomous vehicle's software stack. To achieve our mission, we architect and create high-performance custom silicon; we develop system-level compute architectures that push the boundaries of performance, power, and latency; and we collaborate with many other teammates to ensure we design and improve hardware and software for maximum performance. We are a diverse team looking for curious and talented teammates to work on one of the world's highest performance automotive compute platforms.

In this hybrid role, you will report to a Software Engineering Manager.

You will:

• 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

You have:

• 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

We prefer:

• Python programming experience
• Experience with compilers for neural networks
• Knowledge of computer architectures used for neural network inference, and neural network performance characteristics
• Knowledge of the principles behind popular machine learning and neural network algorithms and applications

#LI-Hybrid