Senior Math Libraries Engineer - LLM Integration and Developer Experience

NVIDIA is a technology company known for its GPU invention and leadership in AI computing, parallel computing for science and engineering, and modern computer graphics. The team develops GPU-accelerated libraries and SDKs used in AI, data analytics, scientific simulations, and other advanced fields.

• PhD or MSc degree in Computational Science, Computer Science, Applied Math, or related field preferred (or equivalent experience).
• 3+ years of experience.
• Robust knowledge of LLMs, finetuning, RL, building RAGs, MCP, and agenting tooling.
• Proven experience designing clear, composable APIs and writing high-quality, well-documented code.
• Ability to prioritize multiple projects and work independently with minimal direction.
• Excellent collaboration, communication, and documentation skills.

Nice to have:

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

• Drive the architecture and implementation of math libraries APIs and documentation to be LLM-first, introspectable, explainable, and synthesizable by AI agents.
• Collaborate 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 transform math libraries' APIs into LLM-friendly representations across popular codegen tools.