← All roles

Senior AI Frameworks Engineer

NVIDIA· Engineer, Sys SW
Apply on company site
Santa Clara, CA, Austin, TX$152k - $241.5kfull-timePosted 5mo ago

About the role

Senior AI Frameworks Engineer (C++/Python) for CUTLASS and CUTLASS DSL, focusing on designing APIs, build/compile infrastructure, and production-grade delivery for open-source GPU programming tools.

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.

Qualifications

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

Skills

Required:PythonC++

Benefits

  • equity
  • benefits (as listed in the job description)
Last updated Jul 3, 2026. Interested? Apply on the company site.

About NVIDIA

World Leader in AI Computing

NVIDIA invents the GPU and drives advances in AI, HPC, gaming, creative design, autonomous vehicles, and robotics.

AIautonomous vehiclesgaminggpugraphics cardhigh performance computingliquid-ai-recruiting-targetllm
Company profileWebsite

More open roles at NVIDIA