Senior AI Frameworks Engineer

NVIDIA

Senior AI Frameworks Engineer at NVIDIA developing Pythonic CUTLASS, a high-performance framework for AI. Bridging low-level hardware and high-level productivity for AI researchers and engineers.

Posted 5/28/2026full-timeSanta Clara • California, Texas • 🇺🇸 United StatesSenior💰 $152,000 - $241,500 per yearWebsite

Tech Stack

Tools & technologies

Python

About the role

Key responsibilities & impact

As a core contributor to the CUTLASS project, you will use your expertise in systems programming and API design to create a world-class developer experience for GPU programming and kernel delivery
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 the user-facing documentation and testing

Requirements

What you’ll need

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.

Benefits

Comp & perks

Equity
Benefits 📊 Check your resume score for this job Improve your chances of getting an interview by checking your resume score before you apply. Check Resume Score

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

PythonC++API designsystems programmingcompilation infrastructureAST transformationsJIT executiondebugging toolsprofiler integrationshigh-performance GPU programming

Soft Skills

user productivitydeveloper experienceintuitive designcommunicationcollaboration

Certifications

MS in Computer SciencePhD in Computer ScienceMS in Electrical EngineeringPhD in Electrical Engineering