Senior Math Libraries Engineer – AI and HPC
NVIDIA
Opportunity published from Workday · applications stay with the original hiring source.
About the opportunity
NVIDIA Math Libraries team is seeking a Senior Engineer to join our development efforts in kernel generation for AI and HPC, focusing on matrix operations, JITing, and fusions.
Our team develops GPU-accelerated mathematical libraries powering commercial and academic organizations revolutionizing AI, scientific and engineering simulations, and data analytics worldwide.
This hybrid role (#LI-Hybrid) involves scoping, designing, and implementing high-performance numerical dense linear algebra software on GPUs, while providing technical leadership and collaborating with product management and internal and external customers.
Key responsibilities
- Scoping, designing, and implementing high quality and performance numerical dense linear algebra software on GPUs.
- Owning the execution of projects involving multiple engineers and sometimes teams.
- Providing technical leadership and feedback to library engineers working with you on projects and sometimes mentor interns.
- Working closely with product management and other internal and external customers 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.
- Finding and explaining complex solutions, exercising leadership, and coordinating with multiple teams to achieve goals.
Qualifications
- PhD, Master's, or Bachelor's degree in Computer Science, Applied Math, or related science or engineering field of study (or equivalent experience).
- 8+ years of experience in designing, developing, testing, maintenance, and performance optimization of HPC software using C++.
- Strong fundamentals in kernel generation and composable library design for linear algebra.
- Leadership skills in driving software development projects.
- Strong collaboration, communication, and documentation habits.
- Kernel generation with JIT focus/experience desired.
Nice to have
- Experience with parallel programming, ideally using CUDA, MPI, OpenMP, OpenACC, pthreads.
- Good understanding of Machine Learning and Deep Learning technologies as well as knowledge of GPU (preferred) or CPU hardware architecture.
- Experience with low level programming using assembly for performance optimization and operator fusion.
- Experience with agile software development practices using project management tools such as JIRA.
- Experience with a scripting language, preferably Python.
Benefits
- Competitive salary package
- Comprehensive benefits
Job details
- Location: Hybrid (#LI-Hybrid)
- Base salary range: 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5
- Application deadline: October 9, 2026
About the company
NVIDIA is widely considered to be one of the technology world's most desirable employers, featuring some of the most forward-thinking and hardworking people in the world.
We are looking for creative and autonomous engineers with a genuine passion for advancing the state of AI and machine learning across a variety of industries.
Additional information
- Base salary is determined based on location, experience, and the pay of employees in similar positions.
- This posting is for an existing vacancy.
- NVIDIA uses AI tools in its recruiting processes.
Application: UpVagas lists this opportunity from the hiring company or its recruitment platform. The application is completed at the original source.
