Software Engineer - Kernels
Core
Design and optimize low-level kernels that interface directly with custom hardware to support large-scale ML workloads for AGI.
Role type
Senior IC kernel engineer (hardware-software integration)
Builds
Optimized compute kernels for ML training and inference
Domain
AI infrastructure / Custom silicon / Systems programming
Deliverable
production ML models
Required skills
Parallelism, SIMD programming, C, assembly-level optimization, GPU/CUDA programming, Rust, C++, Zig
Preferred skills
ML model kernel implementation (Transformers), distributed parallelism (AllReduce, AllToAll, data/tensor parallelism), compiler internals
Technologies
Rust, C, C++, Zig, CUDA, Assembly
Responsibilities
Design and optimize kernels interfacing with hardware, collaborate with ML Research and Hardware Engineering teams, provide hardware architecture expertise from a programmer's perspective
Seniority
Senior, hands-on IC
