NVIDIA Logo

NVIDIA

Senior Math Libraries Engineer - Sparse Linear Algebra

Posted 22 Days Ago
Be an Early Applicant
Remote
2 Locations
184K-357K
Senior level
Remote
2 Locations
184K-357K
Senior level
Develop and optimize high-performance numerical sparse linear algebra software, provide technical leadership, and enhance library performance in a collaborative environment.
The summary above was generated by AI

We are looking for a software engineer for our Sparse Linear Algebra team which develops key technologies and libraries such as cuDSS, cuSPARSE and cuSPARSELt, for scientific computing and deep learning software stacks targeting a range of processors from edge computing to supercomputers. These high-performance libraries provide accelerated linear algebra functions, such as: matrix products, direct and iterative solvers; and are consumed by industrial and research organizations with applications ranging from gaming and Machine Learning to autonomous driving and chip modeling. Does the idea of being at the heart of these projects and applying your knowledge to develop and optimize algorithms which make an impact around the world excite you? If yes, then come and join our team!

What you will be doing:

  • developing and optimizing scalable high-performance numerical sparse linear algebra software such as direct and iterative sparse solvers,

  • providing technical leadership and guidance to library engineers working with you on projects,

  • working closely with product management and other internal and external partners to understand feature and performance requirements and contribute to the technical roadmaps of libraries,

  • finding opportunities to improve library performance and reduce code maintenance overhead through re-architecting.

These issues are by nature complex and will require you to find and explain solutions, exercise leadership, and coordinate with multiple teams to achieve your objectives.

What we need to see:

  • PhD or MSc’s degree in Computational Science, Computer Science, Applied Math, or related science or engineering field of study is preferred (or equivalent experience).

  • 5+ years experience developing, debugging, and optimizing high-performance parallel numerical sparse linear algebra applications on modern computing platforms, preferably with GPU acceleration using CUDA.

  • Excellent C++ programming and software design skills, including functional and performance tests design.

  • Deep understanding of numerical methods, especially, sparse linear algebra algorithms (e.g., multi-frontal factorization, algebraic multigrid (AMG), etc.)

  • Proven experience in leading and completing software development projects.

  • Excellent collaboration, communication, and documentation habits.

Ways to stand out from the crowd:

  • Experience developing libraries consumed by many users.

  • Experience developing distributed memory parallel computing software with MPI or a PGAS library (e.g., NVSHMEM).

  • Good knowledge of compute and network hardware (e.g., Infiniband) architecture.

  • Experience working in an agile software development environment.

  • A scripting language, preferably Python.

NVIDIA’s invention of the GPU in 1999 fueled the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing for science and engineering. More recently, GPU deep learning ignited modern AI — the next era of computing — with the GPU acting as the brain of computers, robots, and self-driving cars that can perceive and understand the world. Today, we are increasingly known as “the AI computing company.” We're looking to grow our company and build our teams with the smartest people in the world.

#LI-Hybrid

The base salary range is 184,000 USD - 356,500 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

You will also be eligible for equity and benefits. NVIDIA accepts applications on an ongoing basis.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.

Top Skills

C++
Cuda
Cudss
Cusparse
Cusparselt
Mpi
Pgas
Python

Similar Jobs

An Hour Ago
Easy Apply
Remote
2 Locations
Easy Apply
164K-226K Annually
Senior level
164K-226K Annually
Senior level
Artificial Intelligence • Fintech • Machine Learning • Social Impact • Software
Design and implement robust API solutions using gRPC, lead API Management strategy, mentor junior engineers, and collaborate with cross-functional teams.
Top Skills: GrpcKafkaKotlinPythonRubyTypescript
An Hour Ago
Remote
Hybrid
2 Locations
135K-250K Annually
Expert/Leader
135K-250K Annually
Expert/Leader
Artificial Intelligence • Cloud • Sales • Security • Software • Cybersecurity • Data Privacy
The Senior Staff DevOps Engineer will lead the Infrastructure Platform Team, design and implement SaaS infrastructure, and automate operations for high availability. Responsibilities include improving cloud architecture, managing cross-functional teams, and ensuring security compliance.
Top Skills: AWSKubernetesPythonRubyShell ScriptingTerraform
An Hour Ago
Remote
12 Locations
Mid level
Mid level
Blockchain • Internet of Things • Payments • Cryptocurrency • Web3
Develop privacy-enhancing technologies for the Chainlink product stack, collaborating on off-chain and on-chain solutions for consumer protection and new use cases.
Top Skills: Fully Homomorphic EncryptionGoPrivacy PoolsRustStealth AddressesTrusted Execution EnvironmentsWasmZero Knowledge Proofs

What you need to know about the Los Angeles Tech Scene

Los Angeles is a global leader in entertainment, so it’s no surprise that many of the biggest players in streaming, digital media and game development call the city home. But the city boasts plenty of non-entertainment innovation as well, with tech companies spanning verticals like AI, fintech, e-commerce and biotech. With major universities like Caltech, UCLA, USC and the nearby UC Irvine, the city has a steady supply of top-flight tech and engineering talent — not counting the graduates flocking to Los Angeles from across the world to enjoy its beaches, culture and year-round temperate climate.

Key Facts About Los Angeles Tech

  • Number of Tech Workers: 375,800; 5.5% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Snap, Netflix, SpaceX, Disney, Google
  • Key Industries: Artificial intelligence, adtech, media, software, game development
  • Funding Landscape: $11.6 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Strong Ventures, Fifth Wall, Upfront Ventures, Mucker Capital, Kittyhawk Ventures
  • Research Centers and Universities: California Institute of Technology, UCLA, University of Southern California, UC Irvine, Pepperdine, California Institute for Immunology and Immunotherapy, Center for Quantum Science and Engineering

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account