Senior Math Libraries Engineer - LLM Integration and Developer Experience jobs in United States
cer-icon
Apply on Employer Site
company-logo

NVIDIA · 1 day ago

Senior Math Libraries Engineer - LLM Integration and Developer Experience

NVIDIA is a leader in AI computing and technology innovation. The company is seeking a Senior Math Libraries Engineer to accelerate the integration of APIs across the CUDA-X math library software stack into modern LLM workflows, ensuring high-performance applications through effective utilization of CUDA-X libraries.

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

Responsibilities

Drive the architecture and implementation of math libraries APIs and documentation structures designed to be LLM-first: introspectable, explainable, and easily synthesized by AI agents
Work with internal and external stakeholders to deliver timely LLM-enhanced library releases
Build metrics, tools and processes to measure impact and quality of LLM/agentic code generation
Prototype tooling and solutions to help transform math libraries’ APIs into LLM-friendly representations across popular codegen tools

Qualification

LLMs knowledgeAPI designCUDA-X familiarityPhDMSc degreeCollaboration skillsCommunication skillsDocumentation skills

Required

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)
3+ years of experience
Robust knowledge of LLMs, finetuning, RL, building RAGs, MCP, and building agenting tooling
Proven experience designing clear, composable APIs and writing high-quality, well-documented code for complex technical domains
Ability to prioritize multiple projects and work independently with minimal direction
Excellent collaboration, communication, and documentation habits

Preferred

Prior work in AI-assisted software engineering, code generation, or programming language design
Familiarity with CUDA-X math library APIs: cuBLAS, cuFFT, cuSOLVER, cuSPARSE, etc

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