Lead Software Engineer, Kernels
Core
Lead a team of kernel engineers to design and implement high-performance compute kernels for specialized AI hardware, partnering with ML and hardware teams to co-design next-generation architectures.
Role type
Senior IC/Lead Kernel Engineer (AI Hardware)
Builds
High-performance compute kernels for specialized AI hardware
Domain
AI Infrastructure / Hardware-Software Co-design
Deliverable
production ML models
Required skills
Team management, kernel optimization, parallelism, SIMD programming, C, assembly-level optimization, GPU/CUDA programming, C++, C, Zig, Rust
Preferred skills
ML model kernel implementation (Transformers), distributed parallelism (AllReduce, AllToAll, data/tensor parallelism), compiler familiarity
Technologies
CUDA, C++, C, Zig, Rust, Assembly
Responsibilities
Manage and mentor kernel engineers, design and optimize hardware-interfacing kernels, collaborate with ML and hardware teams, provide hardware architecture expertise from a programmer's perspective
Seniority
Senior, hands-on IC with management responsibilities
