Senior AI Frameworks Engineer

NVIDIA; Austin (TX); On-site; USD 152,000 - 241,500; Full time

Benefits offered by this job

Equity options

Comprehensive health benefits

Job summary

NVIDIA is searching for a Senior AI Frameworks Engineer to work on the CUTLASS project, enhancing high-performance computing platforms. You will design intuitive APIs and create tools that elevate developer productivity in AI frameworks.

The ideal candidate will possess at least 3 years of experience, a relevant degree, and strong skills in both C++ and Python. The role offers a competitive salary range from $152,000 to $241,500 based on your qualifications.

Qualifications

• At least 3 years of relevant experience in AI frameworks or compiler projects.
• Strong design skills for Python extensions and foreign function interfaces.
• Deep understanding of the Python ecosystem and high-performance compilation.

Responsibilities

• Design user-friendly APIs for scientific computing frameworks.
• Develop robust compilation infrastructure for Pythonic descriptions.
• Optimize experience with debugging tools and profiler integrations.
• Build delivery infrastructure for open-source community package management.

Skills

Proficiency in Python

Proficiency in C++

Experience in library development

Understanding of GPU Architecture

Education

MS or PhD in Computer Science or Electrical Engineering

Job description

We are now looking for a Senior AI Frameworks Engineer (C++/Python)! NVIDIA's high-performance computing platforms are powering the AI revolution across many applications and industries. Within our software stack, CUTLASS stands out as a popular open-source ecosystem dedicated to high-performance math primitives. Since 2017, it has provided the community with C++ template abstractions to implement custom GEMM and related computations efficiently on NVIDIA GPUs.

We are building the next frontier of this ecosystem: Pythonic CUTLASS (CUTLASS DSL). This initiative aims to bring "speed-of-light" performance and powerful abstractions of our stack directly into the Python environment. Join the CUTLASS team and help bridge the gap between low-level hardware primitives and high-level developer productivity. If you are passionate about building elegant, high-performance DSLs and want to empower the next generation of AI researchers and engineers with better tools, apply today!

What You’ll Be Doing

• 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.

What We Need To See

• 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.

Ways To Stand Out From The Crowd

• Active maintainer status or significant contributions to high-performance open-source libraries, AI frameworks or compiler projects (LLVM/MLIR).
• Understanding of compiler foundations, such as intermediate representations (IR), lowering passes, or AST manipulation.
• Experience with GPU Architecture and parallel programming models (CUDA).

Compensation & Benefits

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is $152,000 USD - $241,500 USD for Level 3, and $184,000 USD - $287,500 USD for Level 4. You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until June 1, 2026.

Equal Opportunity Employer Statement

NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.