Senior Software Engineer - Python Numerical Computing Libraries

Job Overview

We are looking for an experienced software professional to contribute to the design and development of accelerated and distributed implementations of Python APIs for numerical computing. Python has become the de‑facto programming language for practitioners in AI, data science, and HPC through popular frameworks such as Num Py, Sci Py, Tensor Flow, and PyTorch. NVIDIA has been at the forefront of providing GPU‑accelerated implementations of the fundamental components of these frameworks.

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

This developer will be a crucial member of a team that is working to unlock the power of distributed GPU computing for domains such as scientific computing, data analytics, deep learning, and professional graphics, running on hardware ranging from supercomputers to the cloud.

• Work closely with product management and internal or external partners to understand use cases and requirements, and 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, especially as part of customer‑critical end‑to‑end workflows.
• 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 underpin the foundation of multi‑GPU computing at NVIDIA.

• 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, dense and sparse array computing.
• Deep familiarity with Python numerical computing libraries (e.g., Num Py, Sci Py), including accelerated implementations (e.g., CuPy, Jax.Num Py, NumS, cu Numeric).
• Experience developing and publishing Python libraries, following standard methodologies for pythonic API design.
• Strong background with parallel programming and performance analysis.

• Experience using/contributing to Python libraries for data science (e.g., Pandas), machine learning (e.g., scikit‑learn), and deep learning (e.g., Tensor Flow, PyTorch).
• Experience with low‑level GPU performance optimization.
• Experience building, debugging, profiling, and optimizing distributed applications on supercomputers or the cloud.
• Background with tasking or asynchronous runtimes.
• Background on compiler optimization techniques, and domain‑specific language design.

The base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is $184,000

USD–$287,500

USD for Level4, and $224,000

USD–$356,500

USD for Level
5. You will also be eligible for equity and benefits. Applications for this job will be accepted until July1,2026.

NVIDIA uses AI tools in its recruiting processes. NVIDIA is committed to fostering an inclusive work environment and is proud to be an equal‑opportunity employer. We do not discriminate on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status, or any other characteristic protected by law.