Senior ML Performance Engineer, Inference Optimisation at Wayve, London

Looking for ML performance engineer jobs in London? Wayve is hiring a Senior ML Performance Engineer, Inference Optimisation to make large transformer models run efficiently on low-cost, low-power in-vehicle compute.

ML Performance Engineer Jobs in London: Overview

About the Employer

Wayve is the leading developer of Embodied AI technology for autonomous driving. Founded in 2017 and headquartered in London, it uses an AI Driver that learns from experience rather than HD maps or hand-coded rules. The company has partnered with Uber for supervised autonomous rides in London.

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
• Comfortable 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

Visa Sponsorship and International Applicants

Relocation support and visa sponsorship are available where applicable. Wayve is a licensed sponsor on the UK register for the Skilled Worker route. International applicants are considered subject to the role and candidate profile. Hybrid working combines time in the London office and workshops with remote work.

Salary and Benefits

• Meaningful equity
• Relocation support and visa sponsorship where applicable
• Hybrid working with core hours
• Learning and development budgets
• Health insurance, dental, enhanced maternity and paternity leave
• Retirement or pension where applicable
• Access to therapists and wellbeing partnerships

How to Apply

Apply through Ashby. Highlight experience with inference optimisation tools such as TensorRT or CUDA.

Apply to this Job

Application Tips

• Prepare examples of profiling and optimising inference performance on edge hardware
• Mention any experience with NVIDIA or Qualcomm platforms

Browse more UK Jobs and Technology Jobs on CMAJobs.