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.

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.

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.

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.

FoodLabs logo
FoodLabs

Senior C++ Computer Vision Engineer

Join a cutting-edge AI-DeepTech startup in Berlin as a Senior C++ Computer Vision Engineer. Work on world-class on-device AI technology.

DeepL logo
DeepL

Senior Backend Engineer C++

Join DeepL as a Senior Backend Engineer C++ to design and maintain scalable backend services using C++ and AI technologies.

Codeium logo
Codeium

Software Engineer, High Performance ML

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

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.

Waymo logo
Waymo

Software Engineer, Reasoning Foundation

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

Together AI logo
Together AI

Senior Backend Engineer - Java, Rust, Go

Join Together AI as a Senior Backend Engineer in Amsterdam. Work with Java, Rust, and Go to build scalable backend systems.

dataroots logo
dataroots

Expert Machine Learning Engineer

Join Dataroots as an Expert Machine Learning Engineer to design and deliver AI-powered solutions, focusing on machine learning models.

yourfirm GmbH logo
yourfirm GmbH

Senior Fullstack Developer for AI-Driven Mission Technologies

Seeking a Senior Fullstack Developer for AI-driven mission technologies, focusing on Java, JavaScript, Python, and C++. Remote work available.

Reaktor logo
Reaktor

Lead Developer with DevOps and Functional Programming

Join Reaktor as a Lead Developer in Amsterdam, focusing on DevOps, Functional Programming, and JavaScript in a hybrid work environment.

Grammarly logo
Grammarly

Entry Level Back-End Software Engineer (Java)

Join Grammarly as an Entry Level Back-End Software Engineer in Berlin. Work with Java, AWS, and more in a hybrid environment.

Applied Intuition logo
Applied Intuition

Software Engineer - Autonomous Driving

Join Applied Intuition as a Software Engineer in Munich to tackle autonomous driving challenges with top ADAS/AV programs.

BCG X logo
BCG X

AI Engineer

Join BCG X as an AI Engineer in Milan, Italy. Develop AI solutions, partner with clients, and drive innovation in a dynamic environment.

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.

Persona logo
Persona

LLM Backend Developer

Join Persona as a LLM Backend Developer, work remotely, and develop AI-driven backend systems for top startups.

Waymo logo
Waymo

Senior Backend Software Engineer

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

Cere Network logo
Cere Network

Principal AI Engineer

Join Cere Network as a Principal AI Engineer to drive AI innovation in Web3. Requires 10+ years in AI/ML, NLP, and software development.

Nebius AI logo
Nebius AI

Senior Software Engineer (C++)

Join Nebius as a Senior Software Engineer (C++) to develop reliable cloud services in a hybrid work environment.

Uber logo
Uber

Staff Software Engineer - Backend

Join Uber as a Staff Software Engineer - Backend, focusing on membership systems. Work with Java, Python, C++, and more in Amsterdam.