Staff ML Performance Engineer Training Efficiency
Wayve is a London-headquartered embodied-AI company developing and licensing mapless, vehicle-agnostic driving software for assisted, automated, and robotaxi applications.
About Wayve
Wayve Technologies Ltd. develops the Wayve AI Driver, an end-to-end, data-trained software platform that runs on onboard vehicle compute and native sensors. It is designed for OEM integration across L1 driver assistance through L4 automated driving, without HD maps.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will profile large-scale machine-learning workloads, identify bottlenecks, and implement improvements to training and inference efficiency. You will build observability and benchmarking tools, track performance metrics, and work with research teams to integrate optimizations.
Requirements
- 10+ years of industry experience in performance engineering across machine-learning systems, GPU infrastructure, distributed platforms, or a similar field
- Experience optimizing large-scale jobs on GPU compute clusters
- Experience working with platform and research teams
- Experience writing, reporting, and tracking performance benchmarks
- Ability to write high-quality, well-structured, tested Python code
- Bachelor's or master's degree in Machine Learning, Computer Science, Engineering, a related technical discipline, or equivalent experience
- Experience with concurrent, parallel, and distributed computing
- Experience using NVIDIA Nsight Systems or other system profilers
- Experience implementing GPU kernels with CUDA or Triton
- Knowledge of computing fundamentals
Responsibilities
- Profile machine-learning workloads to identify bottlenecks
- Design and implement training and inference efficiency improvements
- Build observability tools to track MFU, throughput, and latency
- Build benchmarking tools to track efficiency gains and regressions
- Collaborate with research teams to integrate performance improvements
Benefits
- Competitive equity package
- Hybrid working arrangement