200+ applicants

Company

NVIDIA · 17 hours ago

Senior Software Engineer - Python Numerical Computing Libraries

United States

Full-time

Remote

Senior Level

$148K/yr - $276K/yr

5+ years exp

Maximize your interview chances

Artificial Intelligence (AI)GPU

Growth Opportunities

H1B Sponsor Likely

Insider Connection @NVIDIA

Discover valuable connections within the company who might provide insights and potential referrals.
Get 3x more responses when you reach out via email instead of LinkedIn.

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

Find out how your skills align with this job's requirements. If anything seems off, you can easily click on the tags to select or unselect skills to reflect your actual expertise.

PythonC++CUDANumerical methodsArray computingNumPySciPyParallel programmingPerformance analysisCuPyJax.NumPyNumSCuNumericPython librariesData science librariesMachine learning librariesDeep learning librariesGPU optimizationDistributed applicationsCompiler optimizationDomain-specific languages

Required

BS, MS or PhD degree in Computer Science, Applied Math, Electrical Engineering or related field (or equivalent experience)

5+ 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 and benefits

Company

NVIDIA

Glassdoor

4.6

NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.

Founded in 1993

Santa Clara, California, USA

10,001+ employees

https://www.nvidia.com

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

2023 (735)

2022 (892)

2021 (696)

2020 (534)