Senior AI Frameworks Engineer (C++/Python) Pythonic GPU DSL

NVIDIA; Austin (TX); On-site; USD 152,000 - 241,500; Full time

Benefits offered by this job

Equity options

Comprehensive health benefits

Job summary

NVIDIA is searching for a Senior AI Frameworks Engineer to work on the CUTLASS project, enhancing high-performance computing platforms. You will design intuitive APIs and create tools that elevate developer productivity in AI frameworks.

The ideal candidate will possess at least 3 years of experience, a relevant degree, and strong skills in both C++ and Python. The role offers a competitive salary range from $152,000 to $241,500 based on your qualifications.

Qualifications

• At least 3 years of relevant experience in AI frameworks or compiler projects.
• Strong design skills for Python extensions and foreign function interfaces.
• Deep understanding of the Python ecosystem and high-performance compilation.

Responsibilities

• Design user-friendly APIs for scientific computing frameworks.
• Develop robust compilation infrastructure for Pythonic descriptions.
• Optimize experience with debugging tools and profiler integrations.
• Build delivery infrastructure for open-source community package management.

Skills

Proficiency in Python

Proficiency in C++

Experience in library development

Understanding of GPU Architecture

Education

MS or PhD in Computer Science or Electrical Engineering

Job description

NVIDIA is searching for a Senior AI Frameworks Engineer to work on the CUTLASS project, enhancing high-performance computing platforms. You will design intuitive APIs and create tools that elevate developer productivity in AI frameworks.

The ideal candidate will possess at least 3 years of experience, a relevant degree, and strong skills in both C++ and Python. The role offers a competitive salary range from $152,000 to $241,500 based on your qualifications.