We are seeking a passionate engineer to enhance the integration of APIs within the CUDA-X math library software stack for modern LLM and agentic AI workflows. The team focuses on developing GPU-accelerated libraries and SDKs that support advancements in AI, data analytics, and scientific simulations. The role emphasizes ensuring that both human engineers and AI agents can build high-performance applications efficiently using CUDA-X libraries. If you possess a unique combination of technical expertise and communication skills, we encourage you to apply.

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

• Drive the architecture and implementation of math libraries APIs designed to be LLM-first.
• Collaborate with stakeholders to deliver timely LLM-enhanced library releases.
• Build metrics, tools, and processes to measure the impact and quality of LLM/agentic code generation.
• Prototype tooling and solutions to transform math libraries’ APIs into LLM-friendly representations.

• 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.