Senior Math Libraries Engineer – AI and HPC
NVIDIA· Engineer, Sys SW
Hybrid (Remote)$184k - $356.5kFull-timePosted 7mo ago
About the role
Senior engineer role focusing on kernel generation for AI and HPC, targeting matrix operations, JITing and operator fusion within NVIDIA's Math Libraries team; leadership across projects, collaboration with product and customers, and optimization of dense linear algebra libraries for GPUs.
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 and mentoring interns when applicable.
- Working closely with product management and other internal and external customers to understand feature and performance requirements and contribute to technical roadmaps of libraries.
- Finding opportunities to improve library performance and reduce code maintenance overhead through re-architecting.
- Explain complex solutions, exercise leadership, and coordinate 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 (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. JIT focus/experience desired.
Skills
Required:Kernel generation
Nice to have:JIT (Just-In-Time) compilationParallel programming
Benefits
- Equity and benefits (as part of compensation)
Last updated Oct 9, 2026. Interested? Apply on the company site.
About NVIDIA
World Leader in AI Computing
NVIDIA invents the GPU and drives advances in AI, HPC, gaming, creative design, autonomous vehicles, and robotics.
AIautonomous vehiclesgaminggpugraphics cardhigh performance computingliquid-ai-recruiting-targetllm
More open roles at NVIDIA
- Senior Physical Verification Engineer4d agoEngineer, ASIC
- Senior System Software Engineer, Display Security4d agoEngineer, Sys SW
- Senior FPGA Prototyping Engineer - Hardware4d agoSanta Clara, CA · Engineer, ASIC
- Senior Product Engineer4d agoHybrid (Thailand) · Engineer, Product Development
- Senior Architect - Data Processing Systems, Analytics and Data Intelligence4d agoHybrid (Remote / Local) · Architect