Senior Math Libraries Engineers - Python APIs

Job Description

NVIDIA is seeking a software engineer to design and develop high-performant Python APIs for math libraries, with a focus on GPU accelerated Deep Learning frameworks.

Requirements

• PhD or MSc degree in Computer Science, Applied Math, or related science/engineering field preferred
• Excellent Python and C++ programming and software design skills
• Experience developing Python APIs in pythonic style
• 3+ years experience in developing, debugging, and optimizing high-performance applications on parallel computing platforms
• Strong understanding of numerical methods in science, engineering, and/or deep learning
• Proven experience in working on software development projects in an agile environment

Responsibilities

• Crafting and developing Python APIs for math libraries
• Prototyping integrations of APIs into targeted libraries and frameworks
• Conducting performance analysis, benchmarking, tuning, and optimization of numerical software on various CPU and GPU architectures
• Collaborating with product management and partners to understand requirements and contribute to technical roadmaps
• Improving target user experiences

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