Job Description

NVIDIA is seeking an engineer to enhance the integration of APIs within the CUDA-X math library for modern LLM and AI workflows. The role involves collaborating with stakeholders to deliver LLM-enhanced library releases and ensuring high-performance application development using CUDA-X libraries.

Requirements

PhD or MSc’s degree in Computational Science, Computer Science, Applied Math, or related field preferred
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
Ability to prioritize multiple projects and work independently
Excellent collaboration, communication, and documentation habits

Responsibilities

Drive the architecture and implementation of math libraries APIs designed to be LLM-first
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

Benefits

Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.