Senior Software Engineer, MathDx CuBLASDx And CuSolverDx

at Nvidia
šŸ“ Santa Clara, United States
USD 184,000-287,500 per year
SENIOR
āœ… On-site

SCRAPED

Used Tools & Technologies

Not specified

Required Skills & Competences ?

Python @ 4 CI/CD @ 4 Communication @ 7 Mathematics @ 4 Parallel Programming @ 4 Product Management @ 4 API @ 4 PyTorch @ 4 CUDA @ 4 GPU @ 4

Details

NVIDIA is looking for an expert software engineer to help expand the catalog of Device eXtension (Dx) APIs for its math libraries. The team develops GPU-accelerated mathematical libraries used across HPC and AI. The role focuses on designing, developing, and optimizing math libraries and APIs that integrate with evolving ecosystems (Python, NumPy, JAX, MLIR, etc.) and support hybrid backends and runtime optimization for performance and flexibility.

Responsibilities

  • Design modern, flexible, and easy-to-use APIs for math libraries and lead design reviews with collaborators.
  • Work closely with internal stakeholders (engineering, product management) and external partners (researchers) to understand use cases and requirements.
  • Survey current trends in software systems and become a domain expert.
  • Design and implement high-performance, production-quality code for mathematical libraries targeting GPU acceleration.

Requirements

  • PhD or MSc degree in Computer Science, Applied Mathematics, or a related science/engineering field is preferred (or equivalent experience).
  • 3+ years of experience designing and developing software for high-performance computing and/or AI applications.
  • Advanced C++ skills, including modern design paradigms (e.g., template metaprogramming, RAII).
  • Parallel programming experience with CUDA or OpenCL.
  • Strong collaboration, communication, and documentation habits.

Ways to stand out

  • Experience using graph compilers and/or Just-In-Time compilation workflows (e.g., XLA, LLVM, MLIR, Numba, NVRTC).
  • Programming skills with Python and experience with modern build automation (e.g., CMake) and testing automation (CI/CD, sanitizers).
  • Experience with CCCL, OpenMP, OpenACC, multi-threading, MPI, PGAS.
  • Strong background in numerical methods (e.g., FFT, numerical linear algebra).
  • Experience with scientific and deep learning libraries and frameworks such as PyTorch, JAX, MKL, MAGMA, PETSc, Kokkos.

Compensation and benefits

  • Base salary range listed by level:
    • Level 3: 148,000 USD - 235,750 USD
    • Level 4: 184,000 USD - 287,500 USD
  • You will also be eligible for equity and benefits (see NVIDIA benefits page).

Additional information

  • Applications for this job will be accepted at least until September 23, 2025.
  • NVIDIA is committed to fostering a diverse work environment and is an equal opportunity employer. The company does not discriminate on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status, or any other characteristic protected by law.