Senior ML Engineer — Production AI Performance

EngineersOfAI; Greater London; On-site; GBP 90,000 - 130,000; Full time

Job summary

Wayve is seeking a Senior Machine Learning Engineer to join a high-ownership team delivering production-ready model releases for OEM engagements. This delivery-focused role centers on taking models from training to deployment, working with downstream specialists to ensure on-vehicle viability, and applying practical optimization techniques to meet runtime constraints.

You will ship real systems, iterate quickly, and collaborate across ML and performance engineering teams to balance trade-offs

Qualifications

• 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 TensorRT/CUDA/QNN/Triton/OpenCL and ability to learn adjacent frameworks quickly.
• Comfort operating from high-level model behaviour to low-level kernel/runtime execution.
• Familiarity with model optimisation concepts such as quantisation and distillation.
• Ability to reason across multiple levels of abstraction and communicate trade-offs clearly.
• Strong engineering fundamentals and collaboration skills.

Responsibilities

• Own end-to-end delivery of model releases from requirements through training, evaluation and readiness for deployment.
• Train and iterate on PyTorch models with an experimental approach.
• Debug and improve model performance focusing on latency, memory and bandwidth constraints.
• Apply optimisation techniques such as quantisation and distillation where beneficial.
• Collaborate with ML and performance engineering teams to hand off models and align on priorities.
• Communicate delivery timelines, trade-offs, and readiness criteria to stakeholders.

Skills

Performance optimization

PyTorch

Quantisation

Distillation

Low-level optimization

Collaboration

End-to-end delivery

Tools

TensorRT

CUDA

QNN

Triton

OpenCL

Job description

Wayve is seeking a Senior Machine Learning Engineer to join a high-ownership team delivering production-ready model releases for OEM engagements. This delivery-focused role centers on taking models from training to deployment, working with downstream specialists to ensure on-vehicle viability, and applying practical optimization techniques to meet runtime constraints.

You will ship real systems, iterate quickly, and collaborate across ML and performance engineering teams to balance trade-offs