Senior AI Frameworks Engineer

About the role

We are now looking for a Senior AI Frameworks Engineer (C++/Python) to join the CUTLASS team. The position leverages NVIDIA's high-performance computing platforms and focuses on developing Pythonic CUTLASS, aiming to enhance developer productivity in AI research and engineering.

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.
• Create 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 package distribution, user-facing documentation, and testing.

Requirements

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

Qualifications

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

Skills

• Expertise in systems programming and API design.
• Strong background in Python and C++.
• Experience with GPU programming and parallel computing.
• Knowledge of compiler foundations and IR manipulation.

Benefits

NVIDIA offers competitive compensation packages, including base salaries ranging from $152,000 to $241,500 for Level 3 and $184,000 to $287,500 for Level 4, along with equity and comprehensive benefits.

Pay

Base salary range: $152,000 - $241,500 for Level 3, and $184,000 - $287,500 for Level 4.

Schedule

Full-time position.

Application Instructions

Applications for this job will be accepted at least until June 1, 2026. Please apply through the provided link or contact us for more information.