Senior Math Libraries Engineers - Python APIs

at Nvidia
📍 Santa Clara, United States
USD 148,000-287,500 per year
SENIOR
✅ On-site

SCRAPED

Used Tools & Technologies

Not specified

Required Skills & Competences ?

Software Development @ 4 Python @ 4 Data Science @ 4 TensorFlow @ 4 Communication @ 7 Parallel Programming @ 4 Product Management @ 4 Debugging @ 6 API @ 4 PyTorch @ 4 Agile @ 4 CUDA @ 4 GPU @ 4

Details

NVIDIA is seeking a self-motivated specialist software engineer to design and develop high-performance Python APIs for NVIDIA Math Libraries. These libraries target numerical, scientific, HPC, and AI workloads and must be accessible to the broader Python ecosystem. In this role you will help build and improve APIs that integrate with libraries and frameworks used by the Python-based data science, AI, and scientific computing communities.

Responsibilities

  • Craft and develop Python APIs for math libraries (for example, linear algebra).
  • Prototype integrations of developed APIs into targeted libraries and frameworks.
  • Conduct performance analysis and benchmarking; perform performance tuning and optimization of numerical software on various CPU and GPU architectures.
  • Collaborate closely with product management and internal/external partners to understand feature and performance requirements and contribute to technical roadmaps for NVIDIA math libraries.
  • Identify opportunities to improve target user experiences.

Requirements

  • PhD or MSc degree in Computer Science, Applied Math, or a related science or engineering field is preferred, or equivalent experience.
  • Excellent Python and C++ programming and software design skills, including design of functional and performance tests.
  • Experience developing Python APIs in a pythonic style.
  • Experience using one or more of: NumPy, CuPy, SciPy, JAX, PyTorch, TensorFlow, Numba.
  • 3+ years of experience developing, debugging, and optimizing high-performance applications on parallel computing platforms.
  • Strong understanding of fundamental numerical methods and computations in science, engineering, and/or deep learning.
  • Proven experience working on software development projects in an agile environment.
  • Strong collaboration, communication, and documentation habits.

Ways to stand out

  • Parallel programming experience with GPUs (CUDA or OpenCL), multi-threading, or MPI.
  • Good knowledge of CPU and/or GPU hardware architecture.

Compensation & Benefits

  • Base salary ranges (determined by location, experience, and internal pay bands):
    • Level 3: 148,000 USD - 235,750 USD
    • Level 4: 184,000 USD - 287,500 USD
  • Eligible for equity and company benefits (see NVIDIA benefits).

Additional information

  • Applications for this job will be accepted at least until August 10, 2025.
  • NVIDIA is an equal opportunity employer committed to fostering a diverse work environment.