Nvidia Senior Software Engineer MathDx – cuBLASDx cuSolverDx 

United States, Texas 

266408734

What you'll be doing:

• 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.

What we need to see:

• 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.

Ways to stand out from the crowd:

• 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.