Tech Lead, ML Edge Deployment

Wayve; City Of London; Hybrid; GBP 80,000 - 100,000; Full time

Job summary

A leading AI technology company based in London is looking for a Tech Lead in ML Edge Deployment. You will direct GPU kernel development for deploying advanced driving models in autonomous vehicles. The ideal candidate has strong proficiency in C++, experience in ML frameworks like PyTorch, and the ability to mentor a diverse team of engineers. This full-time role offers a hybrid work environment, combining office and remote work to enhance collaboration and innovation.

Qualifications

• Proven experience as a technical lead or senior engineer on complex engineering projects.
• Excellent interpersonal and communication skills.
• Ability to mentor and guide a team of engineers.

Responsibilities

• Lead a multi-disciplinary team of GPU kernel engineers.
• Deliver critical roadmap milestones for efficient inference on target GPUs.
• Work with ML and Software teams for model optimization.

Skills

Technical lead experience

C++ proficiency

ML frameworks familiarity

Interpersonal skills

Mentorship capabilities

ML deployment pipelines experience

Embedded SoCs knowledge

Tools

PyTorch

Nvidia

Qualcomm

Renesas

Job description

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 the Tech Lead, ML Edge Deployment you’ll lead the strategic direction for Wayve’s GPU kernel development. This role occupies a critical junction between machine learning and embedded systems, enabling us to deploy our transformer-based driving models efficiently onto autonomous vehicles. This is an exciting opportunity to lead in several high impact, early stage projects at Wayve with the ultimate goal of enabling product deployments onto millions of customer vehicles around the world.

Key responsibilities

• You will lead a multi-disciplinary team of GPU kernel engineers, breaking down large milestones into objectives which can be delivered by yourself and the team
• As a hands-on engineer, you will deliver critical roadmap milestones in enabling efficient inference on multiple target GPUs and accelerators
• You will work closely with members of the ML and Software teams to optimise models for deployment on edge
• You will have opportunities to develop new skills, especially in model optimisation

About you

In order to set you up for success as a Tech Lead in ML Edge Deployment at Wayve, we’re looking for the following skills and experience.

• Proven experience as a technical lead or senior engineer on complex engineering projects
• Proficiency in C++ and ML frameworks such as PyTorch
• Excellent interpersonal and communication skills
• Ability to mentor and guide a team of engineers
• Experience with ML deployment pipelines
• Experience with embedded SoCs used in automotive environments, e.g. Nvidia, Qualcomm, Renesas, etc

This is a full-time role based in our office in London. 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. We operate core working hours so you can determine the schedule that works best for you and your team.

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 for this job

To apply, please submit your details through the Wayve careers page.

Wayve UK Demographic Questions

Wayve is committed to creating a diverse and inclusive culture for our employees. It is crucial for us to understand the demographics of our candidate pool to measure our recruitment practices.

There is no requirement for any candidate to answer our demographic questions.

For candidates who complete the questionnaire, their data will be anonymised and used only in the aggregate to inform our attraction strategy. Wayve is an equal opportunity employer and this data will be used for opportunity monitoring purposes.

What is your gender identity? Select...

Which age group do you belong to? Select...

What is your ethnicity? Select...

Do you consider yourself to have a disability or long-term health condition? Select...

What is your sexual orientation? Select...

By checking this box, I consent to Wayve collecting, storing, and processing my responses to the demographic data surveys above.