Machine Learning Engineer, Performance Tooling
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 build self-service AI performance tools across models, hardware targets, and development workflows. You will analyze profiling data, measure and predict performance, identify efficiency losses, monitor regressions, and use data to establish performance targets with engineering stakeholders.
Requirements
- Hands-on performance engineering experience
- Experience with profiling, roofline analysis, latency optimization, and throughput optimization
- Experience owning a tool or service end to end
- Strong Python skills
- Experience profiling and instrumenting production codebases
- Experience developing deep learning models with PyTorch
- Data analysis skills
- Quantitative communication skills
Responsibilities
- Design and build scalable self-service performance tools
- Measure and predict AI performance
- Model theoretical platform performance and identify efficiency losses
- Predict latency, memory, utilization, and compute cost
- Own monitoring and regression alerting for model builds and training runs
- Set performance targets with training and runtime engineers
- Make performance recommendations using data
Benefits
- Hybrid working policy
- Time working from home