Senior Math Libraries Engineer - LLM Integration and Developer Experience

We are looking for a passionate and energized engineer to accelerate the integration of APIs across the CUDA-X math library software stack into modern LLM and agentic AI workflows.

Description

• 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 transform math libraries’ APIs into LLM-friendly representations.

Requirements

• PhD or MSc in Computational Science, Computer Science, Applied Math, or related field preferred.
• 3+ years of experience.
• Robust knowledge of LLMs, finetuning, RL, and building agenting tooling.
• Proven experience designing clear, composable APIs and writing high-quality, well-documented code.
• Excellent collaboration, communication, and documentation habits.