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 4 days ago

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

Staff ML Performance Engineer (Inference Optimisation)
...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-...
Performance
Full time
Work at office
Work from home
Wayve
United Kingdom
4 days ago
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
17 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
28 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
18 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
7 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
27 days 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
14 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
28 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
22 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
Senior Product Manager, Rosso
...Product Lead. Rosso is deeply technical, built on ML-led models and complex engineering covering forecasting, pricing optimisation and portfolio balancing, so this is a role... ...for how well a model or forecast needs to perform, and designing for the edge cases and operational...
Performance
Permanent employment
Full time
Tem
United Kingdom
a month ago
Customer Infrastructure Specialist
...Collaborate with clients to gather necessary information, perform root cause analysis and implement preventative solutions Analyse... ...success rates and performance metrics to identify trends and optimisation opportunities Plan, test, and deploy security and system patches...
Performance
Full time
Focus Group
United Kingdom
26 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
21 days 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
26 days 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
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
17 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
6 days 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
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
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
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
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
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
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
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
Customer Success Manager
...Success Managers play a pivotal role in optimising workflows, analysing customer support... ...sales, customer success, product, and engineering are all aligned to make sure our customers... ...by necessary technology and tools for efficient and effective performance....
Performance
Full time
Remote work
ShiftCare
United Kingdom
a month ago
VP - Rates Sales
...-focused flow-oriented business aligned with client strategy; optimise profitability by leveraging the Group’s customer franchise.... ...precision in all aspects of your work; composure assists optimal performance during challenging circumstances. Advanced proficiency in Microsoft...
Performance
Full time
Work at office
Flexible hours
MUFG Bank, Ltd.
United Kingdom
6 days ago
Senior Software Engineer, Cloud, Simulation
...your career! The role As a software engineer for Wayve’s Simulation Technology team,... ...Challenges you will own Own key performance indicators (KPIs) for simulator cost, SLOs... ...of large-scale machine learning inference systems running in cloud GPU environments...
Performance
Full time
Work at office
Work from home
Wayve
United Kingdom
17 days ago
Embedded Automotive Platforms Software Engineer
...adapt critical AV functions, including inference, navigation, control, localisation, and... ...troubleshoot, and improve software and system performance, ensuring reliability, scalability, and... ...quality processes to maintain high engineering standards. About you In order to...
Performance
Full time
Work from home
Wayve
United Kingdom
3 days ago
Senior React Native Engineer
...continuous learning. We’re now looking for a Senior React Native Engineer to take a leading role in delivering apps that work... ...React Native profiling tools , Sentry crash reporting , and performance optimisation techniques REST API integration expertise using Axios...
Performance
Full time
Remote work
Future
United Kingdom
a month 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!