Senior Math Libraries Engineer – AI and HPC jobs in United States
cer-icon
Apply on Employer Site
company-logo

NVIDIA · 1 day ago

Senior Math Libraries Engineer – AI and HPC

NVIDIA is a leading technology company known for its innovative contributions to AI and HPC. They are seeking a Senior Math Libraries Engineer to develop high-performance numerical software for GPUs, focusing on kernel generation and matrix operations.

AI InfrastructureArtificial Intelligence (AI)Consumer ElectronicsFoundational AIGPUHardwareSoftwareVirtual Reality
check
Growth Opportunities
check
H1B Sponsor Likelynote
Hiring Manager
Bushra S.
linkedin

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
To be successful in your responsibilities which are by nature sophisticated, you will need to find and explain complex solutions, exercise leadership, and coordinate with multiple teams to work towards your goals

Qualification

C++HPC software developmentKernel generationCUDAMachine LearningDeep LearningParallel programmingAgile practicesPythonLeadership skillsCollaboration skillsCommunication skillsDocumentation skills

Required

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. JIT focus/experience desired

Preferred

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 is a huge plus
Experience with agile software development practices using project management tools such as JIRA
A scripting language, preferably Python

Benefits

Equity
Benefits

Company

NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.

H1B Sponsorship

NVIDIA has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1877)
2024 (1355)
2023 (976)
2022 (835)
2021 (601)
2020 (529)

Funding

Current Stage
Public Company
Total Funding
$4.09B
Key Investors
ARPA-EARK Investment ManagementSoftBank Vision Fund
2023-05-09Grant· $5M
2022-08-09Post Ipo Equity· $65M
2021-02-18Post Ipo Equity

Leadership Team

leader-logo
Jensen Huang
Founder and CEO
linkedin
leader-logo
Michael Kagan
Chief Technology Officer
linkedin
Company data provided by crunchbase