Waymo logo

ML Compiler Engineer, Compute

Waymo

Role Overview

Waymo is seeking a talented ML Compiler Engineer to join our Compute Team, which is responsible for delivering the compute platform that runs the autonomous vehicle's software stack. This role involves maximizing the performance of our neural networks by enhancing and extending our production-grade compiler. You will work closely with hardware architects and model developers to develop an understanding of our unique neural network inference platform and neural networks.

Responsibilities

  • Implement compiler support for novel features of our high-performance architecture.
  • Collaborate with other team members to ensure the design and improvement of hardware and software for maximum performance.
  • Develop system-level compute architectures that push the boundaries of performance, power, and latency.

Qualifications

  • BS degree in Computer Science/Electrical Engineering or equivalent practical experience and 3+ years of industry experience.
  • MS degree in Computer Science/Electrical Engineering and 1+ years of industry experience.
  • PhD Degree in Computer Science/Electrical Engineering.
  • 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.

Preferred Skills

  • 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.

Benefits

  • Participation in Waymo’s discretionary annual bonus program.
  • Eligibility for equity incentive plan.
  • Access to a comprehensive company benefits program.

This is a hybrid role based in New York City, reporting to the Manager of Software Engineering. The expected salary range is $158,000—$200,000 USD, depending on the candidate's location, experience, and skills.

Benefits
Extracted with AI

  • Discretionary annual bonus
  • Equity incentive plan
  • Company benefits program

Similar jobs

Last update: 23 minutes ago

Waymo logo
Waymo

Machine Learning Engineer, Training

Join Waymo as a Machine Learning Engineer to develop distributed training infrastructure for autonomous driving.

Qualcomm logo
Qualcomm

Machine Learning Compiler Engineer

Join Qualcomm as a Machine Learning Compiler Engineer to optimize ML compilers for cutting-edge accelerators.

Waymo logo
Waymo

Software Engineer, Commercialization Engineering

Join Waymo as a Software Engineer in Commercialization Engineering, focusing on infrastructure for driverless cars.

Waymo logo
Waymo

Principal Software Engineer, Storage & Compute

Lead software engineer role focusing on storage & compute infrastructure for autonomous driving tech at Waymo.

Wayve logo
Wayve

Senior Machine Learning Performance Engineer

Join Wayve as a Senior Machine Learning Performance Engineer to optimize large-scale training jobs and improve GPU efficiency.

Codeium logo
Codeium

Software Engineer, High Performance ML

Join Codeium as a Software Engineer focusing on high-performance ML, enhancing AI tools for developers.

Waymo logo
Waymo

Software Engineer, Reasoning Foundation

Join Waymo as a Software Engineer in Reasoning Foundation to develop cutting-edge autonomous driving technology.

NVIDIA logo
NVIDIA

Machine Learning Engineer - LLM Fine-tuning and Performance

Join NVIDIA as a Machine Learning Engineer specializing in LLM fine-tuning and performance optimization. Work with cutting-edge ML technologies.

Waymo logo
Waymo

Senior Backend Software Engineer

Senior Backend Software Engineer at Waymo in Warsaw. Develop critical systems for autonomous driving technology.

CentML logo
CentML

Senior Software Engineer - LLM Inference

Join CentML as a Senior Software Engineer focusing on LLM Inference, leveraging AI, ML, and GPU technologies.

Google logo
Google

Early Career Software Engineer, AI/Machine Learning

Join Google as an Early Career Software Engineer specializing in AI/Machine Learning, based in New York. Develop cutting-edge technologies.

The New York Times logo
The New York Times

Software Engineer, Machine Learning Platforms

Join The New York Times as a Software Engineer in Machine Learning Platforms, enhancing digital reader experience through ML. Hybrid work in NYC.

Waymo logo
Waymo

Principal Engineer, Commercialization

Lead Principal Engineer role in Commercialization at Waymo, focusing on backend systems and technical leadership.

Nuro logo
Nuro

Senior Software Engineer, ML Infrastructure

Senior Software Engineer for ML Infrastructure at Nuro, focusing on optimizing ML development and performance.

Mamo logo
Mamo

AI Engineer for Software-Defined Vehicles

AI Engineer role focusing on software-defined vehicles, requiring skills in AI, ML, and cloud technologies.

Mamo logo
Mamo

Senior AI Engineer

Senior AI Engineer role focusing on leveraging AI for automotive software solutions, requiring strong programming and machine learning skills.

Neural Concept logo
Neural Concept

Machine Learning Engineer

Join Neural Concept as a Machine Learning Engineer to solve complex engineering challenges using Python and deep learning.

Snap Inc. logo
Snap Inc.

Machine Learning Engineer

Join Snap Inc. as a Machine Learning Engineer in New York, NY. Develop AI models, collaborate with teams, and drive innovation.

Waymo logo
Waymo

Fullstack Engineer, Applications and Tools

Join Waymo as a Fullstack Engineer in Warsaw, developing applications and tools for autonomous vehicles. Hybrid work model.

Google DeepMind logo
Google DeepMind

Research Engineer, Product

Join Google DeepMind as a Research Engineer to apply cutting-edge AI models to real-world problems. Hybrid work in New York.

Google logo
Google

Software Engineer AI/ML, Devices and Services

Join Google as a Software Engineer AI/ML to develop systems for devices like Pixel and Nest, enhancing supply chain processes.

AMD logo
AMD

Software Machine Learning (ML) Architect

Join AMD as a Software ML Architect to design and implement AI solutions for next-gen GPU products.

Mamo logo
Mamo

AI Engineer for Software-Defined Vehicles

Join Sonatus as an AI Engineer in Boston, focusing on software-defined vehicles, leveraging AI and machine learning.

Tesla logo
Tesla

Software Engineer, Machine Learning Infrastructure

Join Tesla as a Software Engineer in ML Infrastructure to optimize and scale neural network training with Python, C++, and PyTorch.