Senior AI Frameworks Engineer

Responsibilities

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

Languages