This role involves contributing to the design and development of accelerated and distributed implementations of Python APIs for numerical computing. The team focuses on optimizing GPU-accelerated libraries for various ecosystems, supporting applications in scientific computing, data analytics, and deep learning. The developer will work closely with product management and partners to understand requirements and enhance performance across CPU and GPU architectures. This position is crucial for unlocking the potential of distributed GPU computing.

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

• Work closely with product management and partners to understand use cases and requirements
• Architect, prioritize, and develop accelerated and distributed implementations of numerical algorithms
• Design future-proof Python APIs for accelerated numerical and 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 support multi-GPU computing

• Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.