Latest AI JobsBrowse Jobs
wayve logo

Staff ML Performance Engineer

wayveSunnyvale, California USAAI Platform
Apply on company site
Salary
$370,040 – $394,900
Job type
Full-time
Work mode
Remote
Experience
Senior · 10+ yrs
Posted
2 Oct 2026
Status
Open · checked 8d ago

About this role

Based in Sunnyvale, California, the Staff ML Performance Engineer position involves optimizing machine learning workloads across training and cloud inference. This role suits an experienced engineer with a strong background in GPU infrastructure who wants to drive efficiency in autonomous driving technology.

Role Overview

Wayve is hiring a Staff ML Performance Engineer to join the Performance Architecture team within the AI Performance organization. The position focuses on making artificial intelligence workloads faster and more efficient across training and cloud inference to unlock new product capabilities.

Key Responsibilities

Profile machine learning workloads across training and cloud inference using system and kernel-level profilers to locate bottlenecks.
Design and implement efficiency enhancements to maximize MFU, throughput, and utilization through techniques such as parallelism, compilation, mixed precision, and caching.
Build reusable and cross-target optimizations, including custom kernels, data loaders, and frameworks like Triton, rather than relying on one-off fixes.
Develop and implement benchmarking tools to track efficiency gains and prevent regressions on key training and cloud inference workloads.
Partner with platform teams to inform cloud GPU hardware strategy and readiness.
Cultivate a performance optimization culture in collaboration with research and model teams.

Requirements

Ten or more years of industry experience leading performance engineering across machine learning systems, GPU compute infrastructure, distributed platforms, or related fields.
Experience optimizing large-scale workloads on GPU compute clusters for training, inference, or both.
Demonstrated history of writing, reporting, and tracking performance benchmarks openly and accessibly.
Proficiency in writing high-quality, well-structured, and tested Python code.
A Bachelor of Science or Master of Science degree in Machine Learning, Computer Science, Engineering, or a related technical discipline, or equivalent professional experience.

Nice to Have

Experience working with concurrent, parallel, and distributed computing.
Background in optimizing inference serving systems regarding latency, throughput, batching, and caching.
Familiarity with system profilers like NVIDIA Nsight Systems.
Practical experience implementing GPU kernels such as CUDA and Triton.
Strong understanding of computing fundamentals related to fast, secure, and reliable code.

Skills and Tools

Python
GPU compute clusters
Triton
CUDA
NVIDIA Nsight Systems
Distributed platforms

Salary and Benefits

Salary range of $370,040 to $394,900 with equity offerings
Annual market-benchmarked salaries
Relocation support and visa sponsorship where applicable
Hybrid working model with core hours and access to vehicle workshops and labs
Learning and development budgets for training and conferences
Comprehensive benefits including health insurance, dental coverage, enhanced parental leave, pension or retirement plans, therapy access, wellbeing partnerships, and team socials

About the Company

Wayve builds embodied intelligence and a mapless, hardware-agnostic AI platform for autonomous driving. The technology enables vehicles to learn directly from real-world experience, scaling from L2 plus up to L4 automation across global OEM partnerships.

How to Apply

Interested candidates should visit the official Wayve careers page to submit an application.

Skills and tech stack

PythonTritonCUDANVIDIA Nsight SystemsMachine Learning

Questions candidates ask

Where is the position located?

The role is based in Sunnyvale, California USA, with a hybrid working model combining office collaboration and remote flexibility.

What is the salary range for this role?

The salary ranges from $370,040 to $394,900, and the compensation package includes equity offerings.

How many years of experience are required?

Candidates must have 10 or more years of industry experience driving performance engineering across machine learning systems, GPU compute infrastructure, distributed platforms, or similar areas.

Is visa sponsorship available?

Yes, visa sponsorship and relocation support are available where applicable.

What educational background is expected?

A BS or MS in Machine Learning, Computer Science, Engineering, or a related technical discipline is required, or equivalent experience.

Ready to apply?

The application is on the official wayve careers page.

Apply on company site