Senior Math Libraries Engineer – AI and HPC

Job Description

Join NVIDIA's Math Libraries team to work on kernel generation for AI and HPC, focusing on matrix operations, JITing, and fusions.

Requirements

• PhD or Master’s degree in Computer Science, Applied Math, or related science or engineering field of study (or equivalent experience)
• 5+ years of experience in designing, developing, testing, maintenance, and performance optimization of HPC software using C++
• Strong fundamentals in kernel generation and composable library design for linear algebra
• Leadership skills in driving software development projects
• Strong collaboration, communication, and documentation habits

Responsibilities

• Scoping, designing, and implementing high-quality and performance numerical dense linear algebra software on GPUs
• Owning the execution of projects involving multiple engineers and teams
• Providing technical leadership and feedback to library engineers
• Working closely with product management and customers to understand feature and performance requirements
• Finding opportunities to improve library performance and reduce code maintenance overhead

Benefits

• Multiple relocation packages
• Two weeklong shutdowns (mid-summer and year-end) in the US (in addition to PTO)
• 8-week parental leave
• 9 Employee Resource Groups
• Annual bonus offering
• Flexible work arrangements
• Up to 6% 401K matching