Senior AI Frameworks Engineer

NVIDIA is a leading company in high-performance computing platforms driving the AI revolution across various industries. They are seeking a Senior AI Frameworks Engineer to contribute to the CUTLASS project by designing APIs, developing compilation infrastructure, and optimizing developer experience for GPU programming.

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 the user-facing documentation and testing