Senior Machine Learning Engineer, AI Performance
WayveAutonomous Driving company
London, United KingdomSenior
Microsoft
Nvidia
SoftBank
Uber
Mercedes-Benz
Nissan
Data & AI
About the role
TL;DR
Senior Machine Learning Engineer focused on delivering production-ready AI models with tight performance constraints.
- •Wayve is seeking a Senior Machine Learning Engineer to join a high-ownership team focused on delivering production-ready model releases.
- •This role is ideal for engineers who enjoy shipping real systems and iterating quickly, taking models from training to production readiness.
- •Key Responsibilities Own end-to-end delivery of model releases, from requirements to deployment readiness.
- •Train and iterate on PyTorch models with a strong experimental approach.
- •Debug and improve model performance using analytical skills.
- •Apply model optimisation techniques like quantisation and distillation.
- •Collaborate with ML and performance engineering teams for model handoffs and bottleneck identification.
- •Requirements Proven experience improving performance in production systems with tight constraints.
- •Strong hands-on experience training and iterating on deep learning models in PyTorch.
- •Proficiency with at least one relevant stack/toolchain (e.g., TensorRT, CUDA, Qualcomm QNN, Triton, OpenCL).
- •Familiarity with model optimisation concepts such as quantisation and/or distillation.
- •Ability to reason across multiple levels of abstraction, from model behaviour to runtime implications.
Required skills
PythonPyTorch
Nice-to-have skills
Computer Vision
Domain expertise
automotiveai
Tech stack
PythonPyTorch