Senior ML Performance Engineer, Inference Optimisation at Wayve, London

Looking for senior ML performance engineer jobs in the UK with visa sponsorship? Wayve is hiring a Senior ML Performance Engineer, Inference Optimisation in London.

Photo: Waymo autonomous car in San Francisco via Wikimedia Commons, licensed CC BY-SA.

Senior ML Performance Engineer Jobs in the UK: Overview

About the Employer

Wayve is building an end-to-end AI platform for autonomous driving. The company develops mapless, hardware-agnostic systems that learn from real-world experience and partners with global OEMs.

Job Responsibilities

• Profile inference performance across model graphs, compiler/runtime behaviour, kernel execution and memory movement
• Implement and validate optimisations in compilers, runtimes and kernels, including fusion, scheduling and quantisation-aware performance
• Build benchmarking and regression tests to track performance across models, devices and software releases
• Optimise for edge targets such as NVIDIA Orin/Thor and Qualcomm platforms
• Contribute to team tooling, documentation and technical discussions around ML performance
• Work with model developers on performance trade-offs so deployment-aware decisions are made early

Requirements

• Experience improving performance in production or production-adjacent systems with latency, memory, bandwidth, power, thermal or cost constraints
• Strong hands-on experience with at least one relevant stack such as TensorRT, CUDA, Qualcomm QNN, Triton or OpenCL
• Comfort working from high-level model behaviour down to kernel/runtime-level execution
• Strong software engineering fundamentals including debugging, profiling, testing and maintainable code
• Clear communication and ability to work across ML, systems and deployment teams
• Nice to have: experience deploying or benchmarking ML models on embedded or edge devices, familiarity with NVIDIA and/or Qualcomm SoCs, Python and C++ proficiency

Visa Sponsorship and International Applicants

Relocation support and visa sponsorship where applicable. Wayve is a licensed sponsor on the UK register for the Skilled Worker route. International applicants are considered. Sponsorship is not guaranteed for every candidate.

Salary and Benefits

• Hybrid working model
• Learning and development budgets
• Health insurance and dental
• Enhanced maternity and paternity leave
• Retirement or pension where applicable
• Access to therapists and wellbeing partnerships
• Meaningful equity

How to Apply

Apply directly via Ashby. A CV is required. Cover letter is optional.

Apply to this Job

Application Tips

• Highlight hands-on experience with TensorRT, CUDA or similar inference stacks
• Prepare examples of profiling and optimising latency-sensitive systems

Browse more Technology Jobs and Visa Sponsorship Jobs on CMAJobs.