Senior Software Engineer - Python Numerical Computing Libraries

NVIDIA is the world leader in accelerated computing, pioneering solutions in AI and digital twins.
$148,000 - $287,500
Backend
Senior Software Engineer
Remote
5,000+ Employees
5+ years of experience
AI · Enterprise SaaS

Description For Senior Software Engineer - Python Numerical Computing Libraries

NVIDIA is seeking a Senior Software Engineer to join their team developing accelerated and distributed implementations of Python APIs for numerical computing. This role is at the intersection of high-performance computing, AI, and data science, working with popular frameworks like NumPy, SciPy, TensorFlow and PyTorch. The position offers an opportunity to work on GPU-accelerated implementations that power scientific computing, data analytics, and deep learning applications across platforms from supercomputers to cloud systems.

The ideal candidate will have deep expertise in Python numerical computing libraries and parallel programming, with the ability to architect and optimize high-performance implementations. They will work on designing future-proof APIs, improving performance on various architectures, and contributing to the foundation of multi-GPU computing at NVIDIA.

This is a chance to join the world leader in accelerated computing, working on technology that transforms major industries. The role offers competitive compensation including a base salary range of $148,000-$287,500 plus equity, and the flexibility of remote work. The position requires 5+ years of experience and strong skills in Python, C++, and CUDA programming, with a focus on numerical methods and parallel computing.

The team environment is dynamic and collaborative, working closely with product management and partners to deliver solutions that unlock the power of distributed GPU computing. This is an excellent opportunity for someone passionate about high-performance computing and numerical algorithms to make a significant impact in the field.

Last updated an hour ago

Responsibilities For Senior Software Engineer - Python Numerical Computing Libraries

  • Work closely with product management and partners to understand requirements
  • Architect and develop accelerated and distributed implementations of numerical algorithms
  • Design Python APIs for accelerated numerical/scientific computing libraries
  • Analyze and improve performance on CPU and GPU architectures
  • Prototype integrations of developed APIs into targeted frameworks
  • Write maintainable and well-tested production code
  • Contribute to multi-GPU computing runtime systems

Requirements For Senior Software Engineer - Python Numerical Computing Libraries

Python
  • BS, MS or PhD in Computer Science, Applied Math, Electrical Engineering or related field
  • 5+ years of relevant industry experience
  • Excellent Python, C++ and CUDA programming skills
  • Strong understanding of fundamental numerical methods
  • Deep familiarity with Python numerical computing libraries
  • Experience developing and publishing Python libraries
  • Strong background with parallel programming and performance analysis

Benefits For Senior Software Engineer - Python Numerical Computing Libraries

Equity
  • Equity

Interested in this job?

Jobs Related To NVIDIA Senior Software Engineer - Python Numerical Computing Libraries

System Test Design Engineer

Senior System Test Design Engineer position at NVIDIA, focusing on test automation and optimization for network products, requiring 5+ years of experience in software development and hardware testing.

Senior Switch Software Verification Engineer

Senior Switch Software Verification Engineer role at NVIDIA, focusing on Python-based verification infrastructure and test automation for networking systems.

Senior System Test Design Engineer

Senior System Test Design Engineer position at NVIDIA, developing automated tests for networking switches, requiring 5+ years of software development experience and strong programming skills.

Senior ASIC Front End Infrastructure Engineer

Senior ASIC Front End Infrastructure Engineer role at NVIDIA focusing on developing and maintaining core verification and RTL infrastructure for GPU development with competitive compensation and benefits.

Senior System Software Engineer

Senior System Software Engineer role at NVIDIA, focusing on AI for Multimedia components development with C++ and system software expertise required.