Join our dynamic team as a senior software developer to contribute to the design and development of GPU-accelerated and distributed implementations of Python numerical libraries. You will work closely with product management and partners to understand use cases and requirements, architecting and developing accelerated numerical algorithms while designing future-proof APIs for these libraries. Your daily tasks include analyzing and improving performance on various CPU and GPU architectures, prototyping integrations into targeted frameworks, and writing maintainable code. Essential skills include Python, C++, CUDA programming, a strong understanding of numerical methods, and experience with Python numerical computing libraries like NumPy and CuPy. Additionally, familiarity with data science, machine learning, and deep learning frameworks is crucial, along with expertise in parallel programming, performance analysis, and distributed application optimization on supercomputers or the cloud.
Design and develop accelerated and distributed implementations of Python APIs for numerical computing.
Architect and prioritize numerical algorithms to enhance performance on GPU architectures.
Develop future-proof Python APIs for accelerated numerical/scientific computing libraries.
Analyze and optimize the performance of APIs in customer-critical workflows.
Prototype integrations of developed APIs into targeted frameworks like TensorFlow or PyTorch.
Write maintainable, well-tested code for production use in multi-GPU computing environments.
What we're looking for
6+ years of industry experience in software development or equivalent academic background.
Expertise in Python, C++, and CUDA programming languages.
Proficient understanding of numerical methods and array computing (both dense and sparse).
Extensive knowledge of Python numerical libraries like NumPy, SciPy, and their GPU-accelerated counterparts.
Experience developing and publishing Python libraries with standard API design methodologies.
Strong background in parallel programming and performance analysis techniques.
Familiarity with data science, machine learning, and deep learning Python frameworks.
Market check
Salary context
Above market
How this pay compares to similar roles
Similar $180k
This role $236k
$106k
most similar roles pay here
$307k
This role pays more than
90%
of similar roles. Most pay
$142,400–$217,725
— the shaded band above.
At the midpoint, this role pays about
$236k
versus about
$180k
for comparable roles.
Based on 240 similar postings.
Employer
About Nvidia
Nvidia is a leading designer of graphics processing units (GPUs) and system-on-chip units, powering gaming, professional visualization, data centers, and artificial intelligence workloads. Industry: Semiconductors & AI Computing
Nvidia currently has
855 open roles
on FindRole.
Listed pay typically runs
$184,000–$287,500
across 843 roles with salary data.