Engineering Manager, GPU Kernel

Wayve
Full-time
Sunnyvale
Posted on 5 months ago

Job Description

The Engineering Manager for the GPU Kernel team will lead the team responsible for writing custom kernels and libraries to enable transformer-based driving models to run efficiently on embedded GPUs and accelerators. This role involves working closely with ML engineers, software engineers, and researchers to deploy end-to-end AI for autonomous vehicles at scale.

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 optimization

Requirements

  • Proven experience as an Engineering Manager delivering complex engineering projects
  • Experience developing GPU kernels and/or ML compilers
  • Experience optimizing systems to meet strict utilization and latency requirements
  • Excellent interpersonal and communication skills
  • Experience with C++ and ML frameworks such as PyTorch
  • Experience with ML deployment pipelines
  • Experience with embedded SoCs used in automotive environments

Benefits

  • No benefits