Senior Math Libraries Engineer - Sparsity in AI

NVIDIA
Santa Clara
Workplace: HybridFull timeUSD 184,000 - 356,500 annuallyFunction: Data Science & Machine LearningExperience: 6+ yearsEducation: bachelorsSkills: ["Communication","Collaboration","Documentation","Teamwork"]

Lead development of GPU-accelerated C++ libraries to enable unstructured sparsity in DL and HPC, extending support to Python and PyTorch, and optimizing performance across real-world AI workloads. Collaborate with product teams to shape technical roadmaps while delivering production-grade, well-tested code for scalable sparse computations on NVIDIA GPUs.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
NVIDIA
NVIDIA
6 months ago

Senior Math Libraries Engineer - Sparsity in AI

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 5 hours agoStatus: Live

Job Summary

Lead development of GPU-accelerated C++ libraries to enable unstructured sparsity in DL and HPC, extending support to Python and PyTorch, and optimizing performance across real-world AI workloads. Collaborate with product teams to shape technical roadmaps while delivering production-grade, well-tested code for scalable sparse computations on NVIDIA GPUs.
Location: Santa Clara
Workplace: Hybrid
Employment Type: Full time
Job Function: Data Science & Machine Learning

Key Responsibilities

  • •Design and develop a C++-based system to simplify and accelerate computing for unstructured sparsity in DL and HPC on NVIDIA GPUs
  • •Enable the system in languages and frameworks commonly used in DL, such as Python and PyTorch
  • •Evaluate and improve the performance of the system on real-life applications
  • •Realize opportunities to improve library quality, performance and maintainability by writing effective and well-tested code for production use
  • •Work closely with product management and other internal and external partners to understand feature and performance requirements and contribute to technical roadmaps

Pay and Benefits

Salary: USD 184,000 - 356,500 annually
Equity and Bonus:Equity
Perks:Equity

Key Requirements

  • •BS, MS or PhD in Computer Science, Applied Math, or related field (or equivalent experience)
  • •6+ years of experience developing, debugging and optimizing high-performance software using C++ and parallel programming (CUDA, MPI, OpenMP)
  • •Experience with domain-specific language design and compiler optimizations, in particular sparse compilers (MLIR or TACO)
  • •Excellent C++, Python, and CUDA programming skills
  • •Strong collaboration, communication, and documentation habits in a globally distributed organization
Experience:6+ yearsAIGPUHigh-performance computingSparse computationsDLPyTorch
Education:Bachelor's
Skills:CommunicationCollaborationDocumentationTeamwork
Tech Stack:C++PythonCUDAMPIOpenMPMLIRTACOPyTorch

Company Brief

NVIDIA
Designs and manufactures GPUs, AI accelerators, and system-on-chip products for gaming, data centers, professional visualization, and automotive markets, enabling advanced graphics, AI, and high-performance computing solutions worldwide.
Industry: Electronics Manufacturing
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: Santa Clara, United States
Founded: 1993
Glassdoor
Glassdoor: 4.3
WebsiteLinkedInGlassdoor