Senior Compiler Engineer
Core
Build the ML backend (compiler, run-time, debugger) connecting PyTorch, TensorFlow, JAX, and MXNet to custom hardware to maximize AI model performance.
Role type
Senior IC compiler engineer (ML backend)
Builds
Compiler, runtime, and debugger for custom AI accelerators
Domain
AI hardware / High-performance computing
Deliverable
production ML models
Required skills
C/C++ programming, ML framework internals, compiler design, low-level programming, performance optimization, LLVM, OpenXLA/XLA, MLIR, TVM, high-performance computing
Preferred skills
GPU/AI accelerator trends knowledge, prototyping innovations, personal projects
Technologies
PyTorch, TensorFlow, JAX, MXNet, LLVM, OpenXLA/XLA, MLIR, TVM
Responsibilities
Design and implement compiler/runtime/debugger for multiple ML frameworks; Apply advanced optimization techniques (layout, fusion, scheduling, tiling); Mentor peers and lead code reviews; Guide team direction on hardware trends and innovations
Seniority
Senior, hands-on IC with mentorship
