[Remote] Senior Software Engineer, MathDx – cuBLASDx and cuSolverDx

Note: The job is a remote job and is open to candidates in USA. NVIDIA is looking for an expert software engineer to help expand their catalog of Device eXtension APIs for math libraries. The role involves designing, developing, and optimizing math libraries for high-performance computing and AI applications. Responsibilities • Design modern, flexible, and easy to use APIs for math libraries and lead design reviews with all collaborators • Work closely with internal (e.g., engineering, Product Management) and external partners such as researchers to understand their use cases and requirements • Become a domain expert by continuously surveying current trends in software systems Skills • 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 meta-programming, RAII) • Parallel programming experience with CUDA or OpenCL • Strong collaboration, communication, and documentation habits • Experience using graph compilers and/or Just In Time compilation workflows (e.g. XLA, LLVM, MLIR, Numba, NVRTC) • Programming skills with Python, and modern automation setups for both building software (e.g. cmake) as well as testing it (e.g. 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, etc Benefits • Equity and benefits Company Overview • NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI. It was founded in 1993, and is headquartered in Santa Clara, California, USA, with a workforce of 10001+ employees. Its website is Company H1B Sponsorship • NVIDIA has a track record of offering H1B sponsorships, with 1418 in 2025, 1356 in 2024, 976 in 2023, 835 in 2022, 601 in 2021, 529 in 2020. Please note that this does not guarantee sponsorship for this specific role. Apply tot his job