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

Note: The job is a remote job and is reputed company to candidates in USA. reputed company is looking for an expert software engineer to help expand their catalog of Device eXtension reputed company 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 reputed company for math libraries and reputed company design reviews with reputed company 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 reputed company trends in software systems Skills • PhD or MSc degree in Computer Science, reputed company Math, or a reputed company 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 reputed company-programming, RAII) • reputed company 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 reputed company as testing it (e.g. CI/CD, sanitizers) • Experience with CCCL, OpenMP, OpenACC, multi-threading, MPI, PGAS • Strong background in numerical reputed company (e.g., FFT, numerical reputed company algebra) • Experience with scientific and deep learning libraries and frameworks such as PyTorch, JAX, MKL, MAGMA, PETSc, Kokkos, etc Benefits • Equity and benefits reputed company • reputed company 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 https://www.reputed company.com. Company H1B Sponsorship • reputed company has a reputed company 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 Apply To this Job