Want to professionalize your AI skills, pivot to an AI role and increase your salary?
Master AI Engineering with the most practical and comprehensive LLM Development certifications at Towards AI Academy.

Waymo

Tech Lead Manager ML Optimization

Waymo

Published 11 May 2026
Mountain View, CA, USA
298K - 378K USD Annual
Full Time

Share this job

Role Highlights

Languages used

Key skills

Data Engineer
Machine Learning
Model Optimization
Computer Science
Integrations
Technical Leadership
Infrastructure
Perception
Modelling
Deployment
AI
Inference
Transformers
PhD

Tools, Libraries and Frameworks

Flow
JAX
PyTorch
Tensorflow

Description

The role involves leading the development and deployment of large-scale machine learning models using advanced infrastructure. The individual will drive model system codesign practices to meet technical and business requirements while improving efficiency across various platforms. Responsibilities include conducting performance analysis, developing tools for distributed training and inference, and optimizing models for next-generation hardware. The position requires guiding cross-functional efforts to integrate data generation, model development, and deployment pipelines. Additionally, the role entails mentoring junior engineers and managing a team of approximately ten individuals.

Required Qualifications and Skills

The role requires at least ten years of professional software engineering experience, including five years specifically in machine learning infrastructure. Candidates must possess deep knowledge of state-of-the-art machine learning models, architectures, and optimization frameworks. Strong leadership experience in navigating cross-functional teams and managing engineers is necessary. A Master’s or PhD in Computer Science, Engineering, or a related field is preferred.

Disclaimer

Disclaimer: Job and company description information and some of the data fields may have been generated via GPT-4 summarisation and could contain inaccuracies. The full external job listing link should always be relied on for authoritative information.

About the company

Waymo

Description

Waymo is an autonomous driving technology company that began as the Google Self-Driving Car Project in 2009. With a mission to enhance mobility and safety, Waymo has developed the Waymo Driver, powering their fully autonomous ride-hailing and local delivery services. To date, Waymo has achieved over 20 million miles of autonomous driving on public roads across 25 US cities and 20 billion miles of simulation testing. The company emphasizes mission-driven innovation and a collaborative culture, aiming to improve access to mobility while prioritizing safety and community impact.

Share

Share this job

Related jobs

System Design
Reliability
Architecture
Interfaces
Mountain View, CA, USA
Full Time
Product Management
Stakeholder Management
Manager
Data
Mountain View, CA, US; San Francisco, CA, US
Full Time
Data Protection
Risk Management
Program Management
Stakeholder Management
Mountain View, CA, USA
Full Time