Staff ML Performance Engineer (Inference Optimisation)

wayve
London

The role

As a Staff ML Performance Engineer, you’ll play a key role in high-impact projects, optimising ML inference for edge accelerators and GPUs. The focus of this team is to run large transformer-based models efficiently on low-cost, low-power edge devices to enable Wayve’s first driving product.

You’ll help set the technical direction for turning these models into production systems that run reliably on in-vehicle compute. This is a hands-on role working across ML systems, compilers, runtimes, kernels, and embedded deployment, contributing to several early-stage, high-impact projects at Wayve.

Key responsibilities:

  • Profile and pinpoint bottlenecks across the full inference stack (model graph, compiler/runtime, kernel execution, memory movement) and deliver measurable improvements.

  • Implement and validate optimisations in compilers, runtimes, and/or kernels (e.g. operator fusion, scheduling, quantisation-aware performance, custom kernels).

  • Build robust benchmarking and regression testing to ensure performance improvements hold across models, devices, and software releases.

  • Optimise for multiple targets (e.g. NVIDIA Orin/Thor, Qualcomm) and work with teams to support these in a maintainable way

  • Collaborate with model developers to influence architecture and training/deployment decisions that affect on-device performance.

  • Contribute to technical roadmaps and tooling and help raise the standard of performance engineering across the team

About you

Essential

  • Proven experience improving performance in production systems with tight constraints (latency, memory, bandwidth, power/thermal, or cost).

  • Strong proficiency with at least one relevant stack/toolchain (e.g. TensorRT, CUDA, Qualcomm QNN, Triton, OpenCL) and confidence learning adjacent frameworks quickly.

  • Comfort operating at multiple levels of abstraction — from high-level model behaviour down to low-level kernel/runtime execution.

  • Strong software engineering fundamentals (debugging, profiling, testing, and maintainable code).

  • Clear communicator and collaborative teammate; able to align multiple stakeholders on performance trade-offs and priorities.

Desirable

  • Exposure to embedded or edge deployment of ML models, including benchmarking on real devices and handling system-level constraints.

  • Experience with NVIDIA and/or Qualcomm SoCs and performance tooling.

  • Python and C++ proficiency.

  • Experience mentoring others and/or driving technical direction in a small, fast-moving team.

This is a full-time role based in our office in London. At Wayve we want the best of all worlds so we operate a hybrid working policy that combines time together in our offices and workshops to fuel innovation, culture, relationships and learning, and time spent working from home.

#LI-HH1

Posted 2026-07-06

Recommended Jobs

Art Teacher - KS5 Fine Art Specialist - Hackney

Marchant Recruitment
London

Art Teacher – Lead KS5 Fine Art Curriculum and Mentor Portfolio Development for Top Art School Entry – Hackney A high-performing secondary academy in Hackney is seeking an experienced and inspir…

View Details
Posted 2025-11-07

Senior Reinsurance Technician

Harrison Holgate
London

Our client, a City based insurance company, have a new opening for a Senior Reinsurance Technician. This will initially be on a six month contract basis. You will be responsible for handling a caseloa…

View Details
Posted 2026-07-27

Social Worker - Looked After Children

Liquid Personnel
London

Social Worker – Looked After Children – Central London – Salary up to £48,063 Liquid personnel are looking for experienced and hardworking social workers to join the Looked After Childrens team i…

View Details
Posted 2026-06-27

Year 4 Teacher - A Primary School in Islington

Marchant Recruitment
London

Year 4 Teacher – Islington Location: Islington, London Contract: Full-Time, Fixed Term Start Date: January 2026 Salary: Dependent on experience About the School This dynamic, fo…

View Details
Posted 2025-10-15

Teacher of Geography - Enfield - Outstanding School

Marchant Recruitment
Enfield, Greater London

A high-achieving school in Enfield is seeking a passionate Geography Teacher to join their successful Humanities faculty. You will be responsible for delivering engaging lessons across KS3 and KS…

View Details
Posted 2026-05-07

Year 5 Teacher - Independent School - Wandsworth -...

Marchant Recruitment
London

An elite Independent School in Wandsworth is seeking a highly skilled Year 5 Teacher to take on a Full-Time position starting January 2026 . This unique opportunity also includes the potential…

View Details
Posted 2025-10-09

Geography Teacher - Independent School in Kingston...

Marchant Recruitment
Kingston upon Thames, Greater London

Geography Teacher – Independent School in Kingston (January Start) Location: Kingston Start Date: January 2026 Contract Type: Full-time or Part-time, Permanent Salary: Paid to Scale …

View Details
Posted 2025-11-21

KS1 Teacher with SEND Experience - Thriving Mainstream...

Marchant Recruitment
London

A welcoming and inclusive primary school in Westminster is seeking a passionate and experienced SEN Teacher to join their growing Specialist Resource Provision (SRP) for Key Stage 1 pupils. Yo…

View Details
Posted 2025-11-13