jobs

Staff ML Performance Engineer (Training Efficiency)

About us Founded in 2017, Wayve is the leading developer of Embodied AI technology. Our advanced AI software and foundation models enable vehicles to perceive, understand, and navigate any complex environment, enhancing the usability and sa

Company

Wayve

Location

Sunnyvale

Status

Open

Posted

2026-02-26T16:36:07+00:00

Wayve's Training Tech team is hiring a Staff ML Performance Engineer in Sunnyvale, California, on a hybrid schedule. The role centers on profiling large-scale training and inference workloads, finding bottlenecks, and improving efficiency through parallelism, model compilation, mixed precision, observability, and benchmarking. The person will work with research teams to integrate optimizations. The posting asks for 10+ years in performance engineering across ML systems or GPU compute, experience with large jobs on GPU clusters, and strong Python. The salary range is $336,400 to $359,000 plus equity.