Staff ML Performance Engineer (Compiler)

• Take a key role in high-impact projects optimising ML inference for edge accelerators and GPUs. • Run large transformer-based models efficiently on low-cost, low-power edge devices to enable Wayve’s first driving product. • Set the technical direction for turning these models into production systems that run reliably on in-vehicle compute. • Work hands-on across ML systems, compilers, runtimes, kernels, and embedded deployment. • Contribute to several early-stage, high-impact projects within the AI Platform organisation. • Raise the standard of performance engineering across the team. • Work hybrid from the London office.

• Bring proven experience improving performance in production systems with tight constraints across latency, memory, bandwidth, power and thermal, or cost. • Bring strong proficiency with at least one relevant stack or toolchain such as TensorRT, CUDA, Qualcomm QNN, Triton, OpenCL, MLIR, or ONNX, and pick up adjacent frameworks quickly. • Operate comfortably at multiple levels of abstraction, from high-level model behaviour down to low-level kernel and runtime execution. • Bring strong software engineering fundamentals across debugging, profiling, testing, and maintainable code. • Communicate clearly and collaborate well, aligning multiple stakeholders on performance trade-offs and priorities. • Be based in London for this hybrid role.

• Bring experience with compute graph scheduling and execution across multiple targets. • Bring exposure to embedded or edge deployment of ML models, including benchmarking on real devices and handling system-level constraints. • Bring experience with NVIDIA or Qualcomm SoCs and their performance tooling. • Bring Python and C++ proficiency. • Mentor others or drive technical direction in a small, fast-moving team. • Apply even if you do not meet every requirement, as Wayve encourages people passionate about self-driving cars to get in touch. • Let Wayve know if you need any accommodations or adjustments to take part fully in the interview process.

• Identify, implement, and validate optimisations in ML compilers, runtimes, and kernels, including operator fusion, scheduling, quantisation-aware performance, and custom kernels. • Profile and pinpoint bottlenecks across the full inference stack, from model graph through compiler and runtime to kernel execution and memory movement, and deliver measurable improvements. • Build robust benchmarking and regression testing so performance gains hold across models, devices, and software releases. • Develop and optimise for multiple target platforms such as NVIDIA Orin and Thor and Qualcomm, working with cross-functional teams on performant, maintainable solutions. • Collaborate with model developers to influence architecture and training or deployment decisions that affect on-device performance. • Contribute to technical roadmaps and tooling for the team.

Wayve is one of the most exciting AI companies in the UK right now. They are tackling one of the hardest problems in technology, teaching machines to drive, and they are doing it with an approach that the rest of the industry is now converging towards. Backed by SoftBank, Microsoft, NVIDIA, Uber, Mercedes-Benz, Nissan, and Stellantis, Wayve has raised $2.8 billion in total funding and reached a valuation of $8.6 billion. With over 1,000 employees across London, Silicon Valley, Vancouver, Leonberg, Herzliya, and Tokyo, Wayve is scaling fast while keeping its London HQ at the centre. What stands out on Glassdoor (4.4/5 from 112+ reviews) is how consistently employees praise the culture, the calibre of colleagues, and the quality of the technical work. People describe it as some of the most interesting work of their careers. Wayve also offers an on-site chef, private healthcare, competitive pay with equity, and a genuine learning environment where you work alongside world-class ML researchers and engineers. That said, some reviews flag that the pace can be intense and that working across global time zones can stretch working hours. If you are an engineer, researcher, or operator who wants to work on genuinely frontier technology with real-world impact, and you thrive in fast-paced, mission-driven environments, Wayve is a rare opportunity.

• Set technical direction on early-stage projects behind Wayve’s first driving product. • Work across the whole inference stack rather than a single layer of it. • Optimise models for real in-vehicle compute on NVIDIA and Qualcomm hardware. • Work hybrid, combining time in the London office with working from home. • Influence model architecture decisions upstream by working directly with model developers. • Contribute to an inclusive environment that values diversity and new perspectives.

Founded in 2017, Wayve is the leading developer of Embodied AI technology. Its advanced AI software and foundation models let vehicles perceive, understand, and navigate any complex environment, improving the usability and safety of automated driving systems. Wayve builds intelligent, mapless, hardware-agnostic AI products for automakers, aiming to accelerate the transition from assisted to automated driving and create autonomy that propels the world forward.