Waymo

Senior ML Compiler Engineer, Compute

Waymo$213K — $263K *
Information Technology
5 - 7 years of experience
Job Overview by Ladders

Qualifications

  • BS in Computer Science/Electrical Engineering or equivalent experience with 5+ years in industry, OR
  • MS in Computer Science/Electrical Engineering with 3+ years in industry, OR
  • PhD in Computer Science/Electrical Engineering with 1+ year in industry
  • 3+ years working on compilers for parallel architectures
  • 1+ year of experience with ML inference or linear algebra computation
  • C++ programming proficiency, especially with agentic AI acceleration

Responsibilities

  • Leverage agentic AI to enhance software functionality and speed up improvements
  • Analyze and optimize performance of code from our production-grade compiler
  • Design and implement new compiler features for high-performance architecture
  • Collaborate with hardware architects on hardware/software codesign for the inference platform
  • Tune neural networks in collaboration with model developers for improved accuracy and efficiency

Benefits

  • Participation in Waymo's discretionary annual bonus program
  • Eligibility for equity incentive plan
  • Generous company benefits program
  • Hybrid work schedule
  • Influential role in a cutting-edge technology environment
Full Job Description
This role follows a hybrid work schedule and you will report to a Software Engineering Manager.

You will:
  • 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

You have:
  • BS degree in Computer Science/Electrical Engineering or equivalent experience and 5+ years of industry experience OR
  • 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

We prefer:
  • Python programming experience
  • Experience with compilers for neural networks, especially MLIR-based
  • 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


The expected base salary range for this full-time position across US locations is listed below. Actual starting pay will be based on job-related factors, including exact work location, experience, relevant training and education, and skill level. Your recruiter can share more about the specific salary range for the role location or, if the role can be performed remote, the specific salary range for your preferred location, during the hiring process.

Waymo employees are also eligible to participate in Waymo's discretionary annual bonus program, equity incentive plan, and generous Company benefits program, subject to eligibility requirements.

Salary Range

$213,000-$263,000 USD

About Waymo

Waymo LLC is a self-driving technology development company. It is a subsidiary of Alphabet Inc., the parent company of Google. Waymo develops autonomous driving technology and provides ride-hailing services through its Waymo One program. The company has been testing its self-driving technology on public roads since 2009 and has logged millions of miles of autonomous driving. Waymo has partnerships with several automakers and has been working on developing autonomous trucks for use in logistics.
Learn more about Waymo
Size
2,000 employees
Industry
Founded
2009

Similar Jobs

More Jobs at Waymo

More Information Technology Jobs

Find similar Senior ML Compiler Engineer, Compute jobs: