NVIDIA is looking for an experienced software professional to contribute to the design and development of accelerated and distributed implementations of Python APIs for numerical computing, aiming to unlock the power of distributed GPU computing for various domains.
Requirements
- 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), including accelerated implementations (e.g. CuPy, Jax.NumPy, NumS, cuNumeric)
- 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. TensorFlow, PyTorch)
- Experience with low-level GPU performance optimization
Responsibilities
- 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 a 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 underlay the foundation of multi-GPU computing at NVIDIA
Other
- 6+ years of relevant industry experience or equivalent academic experience after BS
- Work closely with product management and internal or external partners, to understand use cases and requirements, and contribute to the technical roadmaps of libraries
- Applications for this job will be accepted at least until December 8, 2025.
- NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer.
- As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) 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.
