NVIDIA · 2 weeks ago
Senior Software Engineer - Python Numerical Computing Libraries
United States
Full-time
Remote
Senior Level
$184K/yr - $288K/yr
6+ years expNVIDIA is a leading technology company focused on accelerated computing and AI. They are seeking an experienced software professional to contribute to the design and development of GPU-accelerated and distributed implementations of Python numerical computing libraries, supporting various ecosystems in scientific computing and data analytics.
AI InfrastructureArtificial Intelligence (AI)Consumer ElectronicsFoundational AIGPUHardwareSoftwareVirtual Reality
Responsibilities
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
Qualification
Required
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
Preferred
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
Benefits
Equity
Benefits
Company
NVIDIA
NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.
10001+ employees
H1B Sponsorship
NVIDIA has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1877)
2024 (1355)
2023 (976)
2022 (835)
2021 (601)
2020 (529)
Funding
Current Stage
Public CompanyTotal Funding
$4.09BKey Investors
ARPA-EARK Investment ManagementSoftBank Vision Fund
2023-05-09Grant· $5M
2022-08-09Post Ipo Equity· $65M
2021-02-18Post Ipo Equity
Leadership Team
Jensen Huang
Founder and CEO
Michael Kagan
Chief Technology Officer
Recent News
2026-01-11
2026-01-11
Company data provided by crunchbase