Senior ML Compiler Engineer - High-Performance AI, Hybrid

Waymo; Mountain View (CA); Hybrid; USD 213,000 - 263,000; Full time

Benefits offered by this job

Discretionary annual bonus

Equity incentive plan

Generous benefits program

Job summary

Waymo is seeking a seasoned software engineer to advance our compute stack for the Waymo Driver. Hybrid schedule with leadership under a Software Engineering Manager, you will shape compiler support, performance tuning, and NN inference on our high-performance hardware.

You will collaborate with hardware and model teams to optimize neural networks, improve compiler features for parallel architectures, and apply agentic AI techniques to production software improvements.

Qualifications

• BS degree in Computer Science/Electrical Engineering or equivalent experience with 5+ years of industry experience.
• MS degree in Computer Science/Electrical Engineering and 3+ years of industry experience
• PhD degree in Computer Science/Electrical Engineering and 1+ years of industry experience
• 3+ years experience working on compilers for parallel architectures
• 1+ years experience working with ML inference or linear algebra computation
• C++ programming skills, including leveraging agentic AI acceleration

Responsibilities

• Leverage agentic AI to accelerate analysis, planning and implementation of production software improvements
• Analyze the performance characteristics of code generated by our production grade compiler, and design and implement optimizations to improve that performance
• Design and implement compiler support for novel features of our high-performance architecture
• Work with hardware architects to understand and influence the development of our unique neural network inference platform through hardware/software codesign
• Work with model developers to tune their neural networks for better inference efficiency and accuracy

Skills

C++ programming

ML inference

Parallel architectures

Education

BS degree in Computer Science/Electrical Engineering

MS degree in Computer Science/Electrical Engineering

PhD degree in Computer Science/Electrical Engineering

Tools

MLIR-based compilers

Python

Job description

Waymo is seeking a seasoned software engineer to advance our compute stack for the Waymo Driver. Hybrid schedule with leadership under a Software Engineering Manager, you will shape compiler support, performance tuning, and NN inference on our high-performance hardware.

You will collaborate with hardware and model teams to optimize neural networks, improve compiler features for parallel architectures, and apply agentic AI techniques to production software improvements.