Senior Software Engineer - Python Numerical Computing Libraries

About this role

We are looking for an experienced software professional to contribute to design and development of accelerated and distributed implementations of Python APIs for numerical computing. In the last decade, Python has become the de-facto programming language for practitioners in AI, data science and HPC, through popular frameworks such as NumPy, SciPy, TensorFlow and PyTorch. These frameworks provide an efficient high-level programming interface, allowing their users to focus on their application while providing highly optimized implementations

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! What you will be doing: 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 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 What we need to see: BS, MS or PhD degree in Computer Science, Applied

Apply for this role on NVIDIA’s official careers site.