Senior Software Engineer, MathDx ā€“ cuBLASDx And CuSolverDx

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 ?

Python @ 4 CI/CD @ 4 Communication @ 7 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 NVIDIA's math libraries. You will be part of a team designing, developing, and optimizing GPU-accelerated mathematical libraries used in HPC and AI workloads, working on modern API design and high-performance implementations.

Responsibilities

  • Design modern, flexible, and easy-to-use APIs for math libraries and lead design reviews with collaborators.
  • Work closely with internal teams (engineering, product management) and external partners (researchers) to understand use cases and requirements.
  • Continuously survey current trends in software systems and become a domain expert.

Requirements

  • PhD or MSc degree in Computer Science, Applied Math, or a related science or 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.

Preferred / 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 automation for building (e.g., CMake) and testing (e.g., CI/CD, sanitizers).
  • Experience with CCCL, OpenMP, OpenACC, multithreading, 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, etc.

Compensation & Benefits

  • Base salary ranges (determined by location, experience, and comparative pay):
    • Level 3: 148,000 USD - 235,750 USD
    • Level 4: 184,000 USD - 287,500 USD
  • Eligible for equity and benefits (link to NVIDIA benefits provided in original posting).

Other Details

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