Staff ML Performance Engineer (Training Efficiency)
WayveSunnyvale, California, United States
Posted February 26, 2026
Full Time
About the Company
Wayve is the leading developer of Embodied AI technology, with a vision to create autonomy that propels the world forward.
Job Description
We are looking for a Staff ML Performance Engineer to join our Training Tech team working on optimizing large scale ML jobs to enable scaling our models to the next order of magnitude.
Requirements
- 10+ years of industry experience driving performance engineering across ML systems, GPU compute infrastructure, distributed platforms or similar field.
- Experience optimizing large scale jobs on GPU compute clusters.
- Experience in working in platform teams and working with research teams.
- Experience in writing, reporting, and tracking performance benchmarks in an open and accessible way.
- Ability to write high quality, well-structured and tested Python code
- BS or MS in Machine Learning, Computer Science, Engineering, or a related technical discipline or equivalent experience