Tech Lead - GPU Kernel Development - Remote

Overview

As the Tech Lead for the GPU Kernel team, you’ll lead the strategic direction for Wayve’s GPU kernel development, enabling efficient deployment of AI models onto autonomous vehicles.

In Short

• Lead a multi-disciplinary team of GPU kernel engineers.
• Deliver critical roadmap milestones for GPU inference.
• Collaborate with ML and Software teams for model optimization.
• Develop new skills in model optimization.
• Work in a hybrid environment combining office and remote work.

Requirements

• Proven experience as a technical lead or senior engineer.
• Experience developing GPU kernels (e.g. CUDA, OpenCL).
• Proficiency in C++ and ML frameworks such as PyTorch.
• Excellent interpersonal and communication skills.
• Ability to mentor and guide a team of engineers.

Benefits

• Inclusive and diverse work culture.
• Opportunities for personal and professional growth.
• Hybrid working policy for flexibility.