Senior Software Engineer - Python Numerical Computing Libraries

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 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. 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 Ways to stand out from the crowd: * 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 * 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 Your 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 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 August 25, 2026. This posting is for an existing vacancy. NVIDIA uses AI tools in its recruiting processes. NVIDIA is committed to fostering an inclusive 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.

Back to blog

Other Jobs To Apply

Amazon Fulfillment Center - Warehouse Packer

iOS Software Engineer, Consumer Engineering

AI Content Editor (Remote) Flexible Hours, $20+/hr

Senior Software Engineer - Client Lifecycle

Work From Home Customer Service Rep

Call Center Specialist [Work From Home - Night Shift - Full Time]

Student Loan Sales Intern (Flexible, Part-Time) | CampusReel

Careermilard Apple Home Advisor: Customer Support $30/H | Apply

Amazon Fulfillment Center - Warehouse Stower

Flexible Part-Time:Flight Attendant-Chicago

Netflix Remote Jobs No Experience (Entry Level, Full Time).Inc

Project Management Expert - Fully Remote | Upto $60/hr

Remote Medical Editor — US Evening Shift

[Work From Home] Weekend, Evening Telehealth LICSW, LMHC

iOS Software Engineer, Consumer Engineering

Sales Rep - Part Time - Flexible Work

Director Measurement

Amazon Fulfillment Center - Coordinator

Licensed Remote Pharmacy Technician

Apple Home Advisor Jobs: Customer Support at Apple

Evening/Weekend Flight Attendant(Chicago)

Online Chat Representative

Customer Service Center Job At Netflix – No Experience Remote Jobs – Amazon Store

Apple At-Home Advisor $37/H – No Experience Needed (Immediate Hiring)

Netflix Careers | Latest Movie Reviewer Job Work From Home | Jobsarabi.com

Full-Time Sales Director (Evening & Weekend Shift) – Remote

Mothers Wanted – Women 21–39 With Prior Pregnancy (Paid $50K–$85K) - Now Hiring

Creative Director (Remote, Fashion, LA)

New Releases Feature Writer | MovieWeb

Remote Data Analyst (Entry Level)

Specialist I Centralized Services WAH

American Car Parts Salesman

Service Desk Analyst - Weekend Work

Stay-at-Home Moms 21–39: Help bless a family - Prior Birth Required Paid $50K-$85K

Pharmacy Technician Implementation Manager - Remote

Audio Transcription Specialist

Part-Time Collections Evening

Part-Time Weekend Remote Customer Service Representative

Part-Time Loan Sales Specialist, Evenings & Weekends (Remote)

Part-Time Licensed Assistant (Health Insurance) | Remote Work

Voiceover Job in Major Sports Brand Ad, Female, Age 25-35 - Women

Remote Walmart Account Operations Lead for Health & Wellness Company

Remote Data Entry Assistant for Teens – Flexible After‑School & Weekend Hours, Earn $13/hr with arenaflex

Netflix Remote Jobs (No Experience From Home) $33/H

Amazon Fulfillment Center - Logistics Associate

Watch Netflix (Tagger): $12-$46

Product Review Tester in Clarksville, IA

Senior Software Engineer - Client Lifecycle

Remote Customer Retention Specialist Part Time Evening

Data Entry Part Time Weekend Jobs (Work From Ho...