Senior AI Frameworks Engineer — Pythonic CUTLASS, GPU

NVIDIA; Santa Clara (CA); On-site; USD 152,000 - 241,500; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is looking for a Senior AI Frameworks Engineer (C++/Python) to enhance their software stack for high-performance computing. You will design intuitive APIs, optimize the developer experience, and develop robust compilation infrastructure for GPU machine code.

The ideal candidate will have a strong background in Python and C++, ideally holding an MS or PhD. NVIDIA offers a competitive salary range of $152,000 to $287,500 based on level and experience.

Qualifications

• 3+ years of experience in relevant fields.
• Experience in library or framework development.
• Deep understanding of Python ecosystem's delivery stack.

Responsibilities

• Design APIs prioritizing user productivity.
• Develop compilation infrastructure for high-performance GPU machine code.
• Create debugging tools and profiler integrations.

Skills

Proficiency in Python

Proficiency in C++

API design

GPU Architecture

Debugging tools development

Education

MS or PhD in Computer Science, Electrical Engineering or related field

Job description

NVIDIA is looking for a Senior AI Frameworks Engineer (C++/Python) to enhance their software stack for high-performance computing. You will design intuitive APIs, optimize the developer experience, and develop robust compilation infrastructure for GPU machine code.

The ideal candidate will have a strong background in Python and C++, ideally holding an MS or PhD. NVIDIA offers a competitive salary range of $152,000 to $287,500 based on level and experience.