Senior AI Frameworks Engineer

NVIDIA is a leading technology company known for its high-performance computing platforms powering the AI revolution. They are seeking a Senior AI Frameworks Engineer to contribute to the CUTLASS project, focusing on enhancing developer productivity in GPU programming through innovative API design and robust infrastructure.

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