Senior AI Frameworks Engineer
Core
Designing Pythonic abstractions (CUTLASS DSL) to bridge low-level GPU hardware primitives with high-level developer productivity for AI research and engineering.
Role type
Senior IC AI Frameworks Engineer (Systems Programming)
Builds
Pythonic CUTLASS ecosystem, high-performance GPU machine code, open-source delivery infrastructure
Domain
High-performance computing, GPU architecture, AI frameworks
Deliverable
production ML models | product features | infrastructure
Required skills
C++ systems programming, Python FFI design, API design, compiler infrastructure (AST/JIT), GPU parallel programming, open-source library development
Preferred skills
CUDA architecture, compiler foundations (IR/lowering), LLVM/MLIR experience, active open-source maintainer status
Technologies
C++, Python, CUDA, CUTLASS, LLVM, MLIR, AST, JIT
Responsibilities
Design developer-centric APIs for GPU programming, develop compilation infrastructure for Python-to-GPU code generation, build debugging and profiling tools, manage open-source package distribution and documentation
Seniority
Senior, hands-on IC
