Software Engineering Manager, Sparse Linear Algebra Libraries

at Nvidia
📍 Santa Clara, United States
USD 224,000-425,500 per year
SENIOR
✅ Hybrid

SCRAPED

Used Tools & Technologies

Not specified

Required Skills & Competences ?

Software Development @ 4 Python @ 4 CI/CD @ 4 Machine Learning @ 7 TensorFlow @ 7 Communication @ 7 Mathematics @ 4 Mentoring @ 4 Parallel Programming @ 4 Performance Optimization @ 4 Jira @ 4 Product Management @ 4 API @ 4 Project Management @ 4 PyTorch @ 7 CUDA @ 4 GPU @ 4

Details

We are looking for a software engineering manager to lead our development efforts for high-performance sparse linear algebra libraries in the AI and scientific computing domains.

Around the world, leading commercial and academic organizations are revolutionizing AI, data analytics, and scientific and engineering simulations, using data centers powered by GPUs. Applications of these technologies include LLMs, computer vision, encryption, computer aided engineering (CAE), electronic design automation (EDA), autonomous vehicles and countless others. Our team develops the GPU accelerated libraries and SDKs that help make these possible.

Responsibilities

  • Lead, mentor, and grow your library engineering team.
  • Be responsible for the quality and performance of your libraries and the planning and execution of projects.
  • Work closely with NVIDIA Research, Developer Technology, and Product Management teams in scientific computing, programming systems, and AI to help collect requirements and contribute to technology roadmaps.
  • Interact with external partners and researchers to understand their use cases and requirements.

Requirements

  • PhD or MSc degree in Computational Science and Engineering, Computer Science, Applied Mathematics, or related science or engineering field (or equivalent experience).
  • 8+ years experience in developing high-performance numerical software.
  • 3+ years experience recruiting, training and leading software engineering teams.
  • Strong fundamentals in numerical methods such as computational sparse/dense linear algebra and direct and iterative solvers for linear systems.
  • Hands-on experience with user facing API design, object-oriented programming, software architecture development, testing, maintenance, and performance optimization of HPC software using C++ and Python.
  • Experience with parallel programming, ideally using CUDA, MPI, OpenMP, OpenACC, pthreads.
  • Strong collaboration, communication, and documentation skills.
  • Experience with software development practices such as CI/CD and project management tools like JIRA.

Preferred Qualifications

  • Experience working in a globally distributed organization.
  • Good knowledge of CPU and/or GPU hardware architecture.
  • Good understanding of Machine Learning and Deep Learning technologies and experience with deep learning frameworks (e.g., PyTorch, JAX, TensorFlow).
  • Strong background in large-scale computing technologies such as PDE solvers, eigenvalue solvers, and time-domain simulation methods (e.g., CFD, FEA).
  • Experience leading and mentoring teams in dynamic environments involving concurrent hardware and software development.

Benefits

  • Competitive base salary.
  • Eligibility for equity and benefits.
  • Opportunity to work at NVIDIA, a leading AI computing company with a diverse and talented workforce.