Senior Software Engineer - Python Numerical Computing Libraries

Job Description

Join NVIDIA's dynamic team to help develop and optimize GPU-accelerated and distributed implementations of Python numerical libraries, supporting Python-based frameworks in various ecosystems.

Requirements

• BS, MS or PhD degree in Computer Science, Applied Math, Electrical Engineering or related field (or equivalent experience)
• 5+ years of relevant industry experience or equivalent academic experience after BS
• Excellent Python, C++ and CUDA programming skills
• Strong understanding of fundamental numerical methods, dense and sparse array computing
• Deep familiarity with Python numerical computing libraries (e.g. NumPy, SciPy)
• Experience developing and publishing Python libraries, following standard methodologies for pythonic API design
• Strong background with parallel programming and performance analysis

Responsibilities

• Work closely with product management and internal or external partners to understand use cases and requirements
• Contribute to the technical roadmaps of libraries
• Architect, prioritize, and develop accelerated and distributed implementations of numerical algorithms
• Design future-proof Python APIs for accelerated numerical/scientific computing libraries
• Analyze and improve the performance of developed APIs on various CPU and GPU architectures
• Prototype integrations of developed APIs into targeted frameworks
• Write effective, maintainable, and well-tested code for production use
• Contribute to the development of runtime systems that underlay the foundation of multi-GPU computing at NVIDIA

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