Senior Software Engineer, MathDx – cuBLASDx and cuSolverDx jobs in United States
cer-icon
Apply on Employer Site
company-logo

NVIDIA · 1 day ago

Senior Software Engineer, MathDx – cuBLASDx and cuSolverDx

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 that support high-performance computing and AI applications.

AI InfrastructureArtificial Intelligence (AI)Consumer ElectronicsFoundational AIGPUHardwareSoftwareVirtual Reality
check
Growth Opportunities
check
H1B Sponsor Likelynote

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

Qualification

C++CUDAHigh-performance computingPythonNumerical methodsGraph compilersCI/CDCollaborationCommunicationDocumentation

Required

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

Preferred

PhD or MSc degree in Computer Science, Applied Math, or a related science or engineering field is preferred (or equivalent experience)
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
Benefits

Company

NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.

H1B Sponsorship

NVIDIA has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1877)
2024 (1355)
2023 (976)
2022 (835)
2021 (601)
2020 (529)

Funding

Current Stage
Public Company
Total Funding
$4.09B
Key Investors
ARPA-EARK Investment ManagementSoftBank Vision Fund
2023-05-09Grant· $5M
2022-08-09Post Ipo Equity· $65M
2021-02-18Post Ipo Equity

Leadership Team

leader-logo
Jensen Huang
Founder and CEO
linkedin
leader-logo
Michael Kagan
Chief Technology Officer
linkedin
Company data provided by crunchbase