Engineering Manager, GPU Kernel

Job Description

Wayve is seeking an Engineering Manager for their GPU Kernel team in London. 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 with ML engineers, software engineers, and researchers to deploy end-to-end AI for autonomous vehicles at scale.

• Leading a multi-disciplinary team of ML GPU kernel engineers.
• Setting key foundational strategy on deployment frameworks, compilers, toolchains and SoCs.
• Setting clear objectives and priorities, and allocating resources efficiently.
• Developing 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.
• Experience with C++ and ML frameworks such as PyTorch (desirable).
• Experience with ML deployment pipelines (desirable).
• Experience with embedded SoCs used in automotive environments, e.g. Nvidia, Qualcomm, Renesas, etc (desirable).

Wayve offers:

• A hybrid working policy combining time in the office and remote work.
• Opportunities to develop new skills.