Engineering Manager, GPU Kernel

Job Description

Wayve is seeking an Engineering Manager for their GPU Kernel team. This role involves leading a team responsible for writing custom kernels and libraries to enable transformer-based driving models to run efficiently on embedded GPUs and accelerators. The Engineering Manager will collaborate closely with ML engineers, software engineers, and researchers to deploy end-to-end AI for autonomous vehicles at scale.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.Responsibilities:

• Lead a multi-disciplinary team of ML GPU kernel engineers.
• Set key foundational strategy on deployment frameworks, compilers, toolchains and SoCs.
• Set clear objectives and priorities, and allocate resources efficiently.
• Develop new skills within end-to-end ML and inference optimisation.

Requirements:

• Proven experience as an Engineering Manager delivering complex engineering projects.
• Experience developing GPU kernels and/or ML compilers (e.g. CUDA, OpenCL, TensorRT, MLIR, TVM, etc).
• Experience optimising systems to meet strict utilisation and latency requirements.
• Excellent interpersonal and communication skills.

Wayve offers:

• A hybrid working policy combining time in the office and remote work.
• Core working hours to allow for flexible scheduling.