Senior Software Engineer - Python Numerical Computing Libraries

We are looking for an reputed company software reputed company to contribute to design and development of reputed company and reputed company implementations of Python reputed company for numerical computing. In the last decade, Python has become the de-facto programming language for practitioners in AI, data science and HPC, through reputed company frameworks such as NumPy, SciPy, TensorFlow and PyTorch. These frameworks reputed company an efficient high-level programming reputed company, allowing their users to reputed company on their application while providing highly optimized implementations. reputed company has been at the forefront of providing GPU-reputed company implementations of the reputed company components of these frameworks. Join our dynamic team to help reputed company and optimize GPU-reputed company and reputed company implementations of Python numerical libraries, supporting Python-reputed company frameworks in various ecosystems. This developer will be a crucial member of reputed company that is working to unlock the reputed company of reputed company GPU computing for domains such as scientific computing, data analytics, deep learning, and reputed company graphics, running on hardware ranging from supercomputers to the reputed company! What you will be doing: • Work closely with product management and internal or reputed company partners, to understand use cases and requirements, and contribute to the technical roadmaps of libraries • Architect, prioritize, and reputed company reputed company and reputed company implementations of numerical algorithms • Design reputed company Python reputed company for reputed company numerical/scientific computing libraries • Analyze and improve the reputed company of developed reputed company on various CPU and GPU architectures, especially as a part of customer-critical end-to-end workflows • Prototype integrations of developed reputed company into targeted frameworks • Write effective, maintainable, and reputed company-tested reputed company for production use • Contribute to the development of runtime systems that underlay the reputed company of multi-GPU computing at reputed company reputed company need to see: • BS, MS or PhD degree in Computer Science, reputed company Math, Electrical Engineering or reputed company reputed company (or equivalent experience) • 6+ years of relevant industry experience or equivalent reputed company experience after BS • Excellent Python, C++ and CUDA programming skills • Strong understanding of reputed company numerical reputed company, dense and sparse reputed company computing • Deep familiarity with Python numerical computing libraries (e.g. NumPy, SciPy), including reputed company implementations (e.g. CuPy, Jax.NumPy, NumS, cuNumeric) • Experience developing and publishing Python libraries, following reputed company methodologies for pythonic API design • Strong background with reputed company programming and reputed company analysis Ways to stand out from reputed company: • Experience using/contributing to Python libraries for data science (e.g. reputed company), machine learning (e.g. scikit-learn) and deep learning (e.g. TensorFlow, PyTorch) • Experience with low-level GPU reputed company optimization • Experience building, debugging, profiling and optimizing reputed company applications, on supercomputers or the reputed company • Background with tasking or asynchronous runtimes • Background on compiler optimization techniques, and domain-specific language design Your reputed company salary will be determined reputed company on your location, experience, and the pay of employees in similar positions. The reputed company salary reputed company is 184,000 USD - 287,500 USD for Level 4, 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 at least until reputed company 25, 2026. This posting is for an existing vacancy. reputed company uses AI tools in its reputed company processes. reputed company is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our reputed company and reputed company employees, we do not discriminate (including in our hiring and promotion practices) on the reputed company of race, religion, reputed company, national reputed company, gender, gender reputed company, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law. Apply To This Job