Senior Staff Software Engineer - Kernels
Core
Developing and optimizing software kernels for next-generation AI hardware compute engines, mapping computational graphs to underlying architectures.
Role type
Senior Staff Software Engineer (AI Kernels & Systems)
Builds
Production software stack for AI compute engines
Domain
AI Hardware / Systems Software / Computer Architecture
Deliverable
production ML models
Required skills
C/C++, Python, Linux, Computer Architecture, Data Structures, Machine Learning Fundamentals, CUDA, FPGA/DSP/GPU/AI Accelerator programming, ML Operators (GEMM, Conv, BLAS, SIMD), Embedded SIMD (Tensilica)
Preferred skills
Startup/Incubation experience, ML Frameworks (TensorFlow, PyTorch), ML Compilers (MLIR, LLVM, TVM, Glow), Deep Learning Models (CV, NLP, RecSys), Cloud Provider/AI Compute Company experience
Technologies
CUDA, Tensilica, TensorFlow, PyTorch, MLIR, LLVM, TVM, Glow
Responsibilities
Develop, enhance, and maintain software kernels for AI hardware; map algorithms to hardware architectures; optimize hardware-software co-design; build compiler infrastructure; implement ML operators and algorithms for specialized hardware.
Seniority
Senior Staff, hands-on IC with leadership
