NVIDIA
Senior AI Frameworks Engineer
$152K–$288K
US, CA, Santa Clara +1 · Senior · Full-time · Posted May 28, 2026
- Location
- US, CA, Santa Clara
US, TX, Austin - Experience
- At least 3+ years of relevant experience.
- Education
- MS or PhD in Computer Science, Electrical Engineering, or related field (or equivalent experience).
About the job
Develop Pythonic CUTLASS DSL and its APIs, compilation, debugging, and delivery infrastructure to enable productive high-performance GPU programming.
What you'll do
- Design intuitive APIs and develop compilation infrastructure that lowers Python descriptions into GPU machine code.
- Create debugging, profiling, and validation tools for kernel development.
- Build open-source delivery infrastructure, including package distribution, documentation, and testing.
What they ask for
- MS or PhD in Computer Science, Electrical Engineering, or a related field (or equivalent experience).
- At least 3+ years of relevant experience.
- Strong Python and C++ proficiency, including Python extensions and foreign function interfaces (FFI).
- Library or framework development experience and deep understanding of the Python delivery stack for compiled extensions.
Nice to have
- Active maintenance or significant contributions to high-performance open-source libraries, AI frameworks, or compiler projects (LLVM/MLIR).
- Knowledge of compiler foundations or experience with GPU architecture and parallel programming models (CUDA).
Pay
152,000 USD - 241,500 USD (base salary, Level 3); 184,000 USD - 287,500 USD (base salary, Level 4)
Tools
CUTLASS, Python, C++, LLVM/MLIR, CUDA, AST, JIT
AI focus
Develop Pythonic CUTLASS DSL tools for high-performance GPU programming, including compilation, debugging, and kernel delivery.
Model-written summary · the original posting is authoritative