Staff ML Performance Engineer (Inference Optimisation)

Full-time

Wayve

About Us
Founded in 2017, Wayve is the leading developer of Embodied AI technology. Our advanced AI software and foundation models enable vehicles to perceive, understand, and navigate any complex environment, enhancing the usability and safety of automated driving systems.
Our vision is to create autonomy that propels the world forward. Our intelligent, mapless, and hardware-agnostic AI products are designed for automakers, accelerating the transition from assisted to automated driving.
In our fast-paced environment big problems ignite us—we embrace uncertainty, leaning into complex challenges to unlock groundbreaking solutions. We aim high and stay humble in our pursuit of excellence, constantly learning and evolving as we pave the way for a smarter, safer future.
At Wayve, your contributions matter. We value diversity, embrace new perspectives, and foster an inclusive work environment; we back each other to deliver impact.
Make Wayve the experience that defines your career!
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.
Wayve is committed to creating an inclusive interview experience. If you require any accommodations or adjustments to participate fully in our interview process, please let us know.
We understand that everyone has a unique set of skills and experiences and that not everyone will meet all of the requirements listed above. If you’re passionate about self-driving cars and think you have what it takes to make a positive impact on the world, we encourage you to apply.
At Wayve we're committed to creating a diverse, fair and respectful culture that is inclusive of everyone based on their unique skills and perspectives, and regardless of sex, race, religion or belief, ethnic or national origin, disability, age, citizenship, marital, domestic or civil partnership status, sexual orientation, gender identity, veteran status, pregnancy or related condition (including breastfeeding) or any other basis as protected by applicable law.
For more information visit Careers at Wayve.
To learn more about what drives us, visit Values at Wayve
For US candidates only, please visit E-Verify Notice and Participation and Right to Work
DISCLAIMER: We will not ask about marriage or pregnancy, care responsibilities or disabilities in any of our job adverts or interviews. However, we do look to capture information about care responsibilities, and disabilities among other diversity information as part of an optional DEI Monitoring form to help us identify areas of improvement in our hiring process and ensure that the process is inclusive and non-discriminatory.

Apply

Vacancy posted 1 day ago

Similar jobs that could be interesting for youBased on the Staff ML Performance Engineer (Inference Optimisation) in United Kingdom vacancy

Staff ML Engineer, Gaia
...synthetic scenarios, including rare or safety-critical events. As a Staff ML Engineer on Gaia, you’ll own and drive work on training and improving... ...modern ML stacks (e.g., PyTorch), including debugging and performance/reliability-minded development Relevant industry...
Performance
Full time
Work at office
Work from home
Wayve
United Kingdom
13 days ago
Senior Machine Learning Engineer, AI Infrastructure, Autonomy
...looking for a full-time Machine Learning Engineer, with deep knowledge and strong... ...models and accelerating model training/inference. Our mission is to solve the autonomous... ...accelerate model iteration and improve performance. Qualifications PhD in CS/CE/...
Performance
Full time
Contract work
Local area
Rivian
United Kingdom
24 days ago
Machine Learning Engineer (Synthetic Data)
...of this role is to build, scale, and optimise next-generation world model architectures... ...for real driving. You sit between ML research and engineering: collaborating with scientists on... ...generation loop: config → large-scale GPU inference → training-ready artefacts, with...
Suggested
Full time
Work at office
Work from home
Wayve
United Kingdom
14 days ago
Machine Learning Engineer, ADAS
...that defines your career! ** ️ About our Engineering Teams** Wayve’s ADAS engineering teams... ...models, to iterating quickly based on performance gaps. The team mixes “online” (on-car,... ...evaluation signals. You’ll work across the full ML lifecycle (data → training → evaluation...
Performance
Internship
Work at office
Remote work
Visa sponsorship
Relocation package
Flexible hours
Shift work
Wayve
United Kingdom
3 days ago
Staff Software Engineer, ML Training and Inference Infrastructure
...future generations. Role Summary As a Staff Software Engineer, ML training and inference infrastructure, you will be a member of the Perception... ...models; and optimizing the training and inference performance. Responsibilities Optimize the...
Performance
Full time
Contract work
Local area
Rivian
United Kingdom
24 days ago
Senior Data Engineering Manager
...team composed of Data Scientists, Data Engineers and ML Ops Engineers, working in the Autonomy... ...improve machine learning model performance across the End-to-End Autonomy Data pipeline... ...studies, evaluation, deployment, and inference optimization. ~ Experience with rigorous...
Performance
Full time
Temporary work
Work at office
Flexible hours
Tesla
United Kingdom
23 days ago
Senior Data Analyst & AI Engineer
...return accurate, contextually aware results. You are the steward of how well our BI tool thinks. Partner with data engineering to optimise performance, and build what's missing Own and maintain the semantic data layer and transformation logic so that dashboards are...
Performance
Full time
Primer
United Kingdom
a month ago
Analytics Engineer - Product Intelligence
...Intelligence function covers the full analytical picture - product behaviour, GTM performance, commercial data science, financial reporting, and operational intelligence. The Analytics Engineers who serve these teams own the data modelling layer that all of it runs on:...
Performance
Full time
Pleo
United Kingdom
10 days ago
International Standardization Lead (Staff Researcher)
£141k - £148k per year
...improve the safety of Waymo’s fully autonomous driving technology. Our experts develop safety goals and strategies, and conduct safety engineering analyses to ensure safety is being considered throughout the design and development of our vehicles. The team develops and...
Full time
Tesla
United Kingdom
18 days ago
Staff GRC Analyst
...About the role We're looking for a Staff GRC Analyst to join our Information Security... ...Compliance, Procurement, Legal, Finance, and Engineers. Our team is highly collaborative and... ...backlog and task assignment in order to perform at your best. How you’ll develop in this...
Full time
Afternoon shift
Pleo
United Kingdom
17 days ago
Staff Data Scientist, Pricing
...Destination We're looking for a Staff Data Scientist, Pricing to... ...prescriptive by building the optimisation layer that turns forecasts... ...full lifecycle, from feature engineering and model selection through to... .... ~ Fluency in the Python ML stack alongside strong SQL...
Full time
Airalo
United Kingdom
23 days ago
Freelance Strategist
...and creator casting logic to content formats, messaging, and performance KPIs Work cross-functionally with internal teams to craft... ...and audience data to inform targeting, creative direction, and optimisation Partner with Paid Media and Client Solutions to interpret...
Performance
Freelance
Kyra
United Kingdom
a month ago
Country Manager
...positive: defining strategy, leading a local sales team, and owning performance end-to-end. You will operate as the local CEO of the... ...and pipeline management Sales process design and optimisation Define KPIs and performance metrics Team Building Hire...
Performance
Full time
Freelance
Local area
Remote work
RealAdvisor
United Kingdom
a month ago
Content Strategist (Account Strategist, Content & SEO)
...initial client strategy, but will also use our clients overall marketing goals, current and ongoing campaigns, and historical performance to optimise and revise strategies over the lifetime of the client while adhering to Directive best practices. Strategists will work...
Performance
Full time
Directive
United Kingdom
a month ago
Growth Marketing Manager
...ll bring to the role ~4–7+ years' experience in Growth or Performance Marketing, with demonstrable ownership of an end-to-end product... ...other digital channels, with experience shaping strategy and optimising performance. ~ Experience building and optimising...
Performance
Full time
Remote work
Prolific
United Kingdom
a month ago
Contract Product Manager
...customer-facing capabilities, partnering closely with engineering, design, data and business stakeholders to deliver... ..., reviews and retrospectives. Measure product performance through KPIs and continuously optimise products based on customer feedback and data insights...
Performance
Contract work
Temporary work
Remote work
AND Digital
United Kingdom
22 days ago
Team Lead - Evergreen Product Management, Private Capital
...product management, ensuring the team operates as a cohesive, high-performing unit with clear priorities and strong cross-functional... ...senior leadership. Drive process standardisation, workflow optimisation and automation to support continued platform growth and scalability...
Performance
Full time
Worldwide
Flexible hours
EQT Ventures
United Kingdom
2 days ago
Design Engineer
...Title Design Engineer (Permanent) Location Warwick, Warwickshire, UK (Onsite) Hours 40/week About Raeon Custom... ...between, we build batteries differently to enable high performance, affordable optimisation. We’re at the start of our journey. We have experience...
Performance
Permanent employment
Full time
Flexible hours
Tesla
United Kingdom
13 days ago
Fuels Manager
...Manager, you'll take ownership of the planning, allocation and optimisation of fuel-grade waste disposal across the Biffa Group.... ...Partner with the Fuels team and wider business to support contract performance, cost control, disposal strategy and the rollout of the Fuels...
Performance
Full time
Contract work
Temporary work
Work at office
Biffa
United Kingdom
a month ago
Content Manager
...and priorities. Identify opportunities for content reuse, optimisation and production efficiency across brands and categories. Briefing... ...partners, including briefing, delivery oversight and performance management. Build strong working partnerships across international...
Performance
Contract work
Temporary work
Inspired Thinking Group
United Kingdom
a month ago
Senior Software Engineer
...Security is seeking an experienced Senior Software Engineer to join the Service Delivery team. The... ...and external customers. Critically analyse and optimise our services and frameworks to ensure performance at scale. Deliver code in a repeatable and scalable...
Performance
Full time
Work at office
Remote work
Menlo Security
United Kingdom
a month ago
Azure Cloud Engineer (Terraform & DevOps)
...The role We're looking for an experienced Azure Cloud Engineer with deep Terraform and DevOps expertise to join our growing... ...networking and landing zones, and you'll monitor, troubleshoot and optimise the performance, reliability and cost of Azure workloads. As the senior...
Performance
Full time
Remote work
Flexible hours
Rotating shift
Focus Group
United Kingdom
a month ago
Senior Expert Acquisition Lead
...scale the acquisition strategies, sourcing engines and systems that enable us to attract,... ...through experimentation, conversion optimisation and data-driven decision making. ~ Strong... ...metrics to evaluate channel performance, funnel health and growth opportunities...
Performance
Full time
Remote work
Prolific
United Kingdom
a month ago
Site Manager
...bodies to ensure environmental compliance, optimise landfill gas and leachate systems, and... ...while maximising operational performance. Your core responsibilities Manage... ...collaboratively with operational, environmental, engineering, compliance, finance and health &...
Performance
Full time
Temporary work
For contractors
Local area
Work from home
Night shift
Biffa
United Kingdom
a month ago
Manager / Senior Manager - Asset & Operations Excellence
...based and predictive maintenance strategies driving improvements in asset performance across the whole asset lifecycle. You'll lead providing advisory services related to and optimisation of EAM / CMMS platforms such as IBM Maximo – shaping user requirements,...
Performance
Full time
Shift work
Baringa
United Kingdom
a month ago
Head of Performance Marketing
...cannabis clinic. With over 170 staff, we connect patients with specialist... ...We're now looking for a Head of Performance Marketing to own our paid acquisition engine and affiliate channel, driving... ...existing affiliate operation, optimising performance, deepening partner relationships...
Performance
Full time
Flexible hours
Releaf
United Kingdom
a month ago
Digital Engineering Manager
£1,500 per month
...The Role Joining the Digital Engineering function would see you work as part of a cross... ...satisfaction, quality and team member performance. Dealing with obstructions and blockers... ...performance improvement and process optimisation to ensure maximum productivity...
Performance
Full time
Inspired Thinking Group
United Kingdom
a month ago
Senior Growth Manager
...LemFi is the financial engine for people the world wasn't built to serve — and nowhere is that more true than for... ...acquisition strategies, using data and AI-assisted tools to optimise campaigns and improve KPI performance Identify, negotiate, and manage partnerships with...
Performance
Full time
Local area
Immediate start
Remote work
LemFi
United Kingdom
a month ago
Senior Software Engineer
...The Role As a Senior Software Engineer in the Acquisition Squad, you will be the driving... ...engineers to build features and products that optimise our signup funnels, referral loops, and... ...of how backend engineering and system performance directly impact core business metrics,...
Performance
Full time
Butternut Box
United Kingdom
a month ago
Data Director - Payments
£150k - £200k per year
...areas. Data Scientists , Analytics Engineers , Machine Learning Engineers and Analysts... ...untapped opportunities for commercial optimisation, including across pricing,... ...decisions Develop and scale a high-performing team of data professionals sharing accountability...
Performance
Full time
Work at office
Local area
Relocation
Visa sponsorship
Flexible hours
Monzo
United Kingdom
20 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Staff ML Performance Engineer (Inference Optimisation). Be the first to apply!