Senior AI Frameworks Engineer: High-Perf GPU DSL + Equity

NVIDIA Corporation; Santa Clara (CA); On-site; USD 180,000 - 287,500; Full time

Job summary

NVIDIA Corporation in Santa Clara, CA, seeks a Senior AI Frameworks Engineer (C++/Python) to advance CUTLASS and Pythonic CUTLASS. You will design APIs, build Python extensions, and deliver high-performance GPU code with a focus on developer experience.

Candidates have MS or PhD, 3+ years in library or framework dev, strong Python and C++, LLVM/MLIR familiarity, and CUDA knowledge. Hybrid/work flexible; equity and benefits apply.

Qualifications

• MS or PhD degree in Computer Science, Electrical Engineering, or related field (or equivalent experience).
• At least 3+ years of relevant experience.
• Strong proficiency in Python and C++, specifically regarding the design of Python extensions and foreign function interfaces (FFI).
• Experience in library or framework development, with a focus on creating intuitive APIs for complex technical systems.
• Deep understanding of the Python ecosystem’s delivery stack, including building, testing, and distributing high-performance compiled extensions.

Responsibilities

• Design APIs that prioritize user productivity, providing a native feel for developers accustomed to modern scientific computing and deep learning frameworks.
• Develop robust compilation infrastructure—including AST transformations and JIT-friendly execution—to lower Pythonic descriptions into high-performance GPU machine code.
• Optimize developer experience by creating debugging tools, profiler integrations, and validation methodologies that make writing and using kernels easy.
• Build production-grade delivery infrastructure for the open-source community, managing everything from package distribution (wheels, conda) to user-facing documentation and testing.

Skills

Python

C++

FFI API design

Open-source

Education

MS or PhD in CS/EE

Tools

LLVM

MLIR

CUDA

Job description

NVIDIA Corporation in Santa Clara, CA, seeks a Senior AI Frameworks Engineer (C++/Python) to advance CUTLASS and Pythonic CUTLASS. You will design APIs, build Python extensions, and deliver high-performance GPU code with a focus on developer experience.

Candidates have MS or PhD, 3+ years in library or framework dev, strong Python and C++, LLVM/MLIR familiarity, and CUDA knowledge. Hybrid/work flexible; equity and benefits apply.