Tech Lead, ML Edge Deployment

Job Description

Wayve is seeking a Tech Lead for ML Edge Deployment to spearhead the strategic direction of the company's GPU kernel development. This role is crucial for efficiently deploying transformer-based driving models onto autonomous vehicles. The Tech Lead will lead a multi-disciplinary team of GPU kernel engineers, breaking down large milestones into achievable objectives. They will also be a hands-on engineer, delivering critical roadmap milestones for efficient inference on various target GPUs and accelerators. Collaboration with ML and Software teams will be essential to optimize models for edge deployment. This role offers opportunities to develop new skills, particularly in model optimization.Role involves:

• Leading a team of GPU kernel engineers.
• Delivering critical roadmap milestones.
• Optimizing models for deployment on edge.
• Collaborating with ML and Software teams.

Requirements:

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

Wayve offers:

• A hybrid working policy.
• Core working hours.