Senior Software Engineer, cuBLASDx and cuSolverDx
NVIDIA
PythonPyTorch
Senior Software Engineer – cuBLASDx and cuSolverDx
NVIDIA
full-time
Posted on:
Origin: • 🇺🇸 United States
Visit company websiteJob Level
Senior
Tech Stack
PythonPyTorch
About the role
- Design modern, flexible, and easy to use APIs for math libraries and lead design reviews with collaborators
- Work closely with internal and external partners (engineering, Product Management, researchers) to understand use cases and requirements
- Continuously survey current trends in software systems and become a domain expert
- Design, develop, and optimize math libraries for future high-performance computing and AI applications
Requirements
- PhD or MSc degree in Computer Science, Applied Math, or a related science or engineering field 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: Experience using graph compilers and/or JIT compilation workflows (XLA, LLVM, MLIR, Numba, NVRTC); Programming skills with Python; Modern automation for build/test (cmake, CI/CD, sanitizers); Experience with CCCL, OpenMP, OpenACC, multi-threading, MPI, PGAS; Strong background in numerical methods; Experience with scientific and deep learning libraries/frameworks (PyTorch, JAX, MKL, MAGMA, PETSc, Kokkos)
