Senior Math Libraries Engineer - LLM Integration and Developer Experience
$152k - $288k • US, CA, Santa Clara
Posted 1mo ago
Job Location
US, CA, Santa Clara
Tech Stack
Remote Work Policy
On-site
Categories
LLM Engineer
About the job
We are seeking a motivated engineer to enhance the integration of APIs across our CUDA-X math library software stack into modern LLM and agentic AI workflows. Organizations worldwide leverage GPU-powered data centers for AI, data analytics, and scientific simulations, driving progress in fields like LLMs, computer vision, and autonomous vehicles. Our team develops the essential GPU-accelerated libraries and SDKs for these technologies. In this role, you will ensure that human engineers and AI agents can easily build high-performance applications by effectively utilizing CUDA-X libraries. We are looking for individuals with a unique combination of technical expertise, developer experience, and communication skills.
Responsibilities
- Drive the architecture and implementation of math library APIs and documentation structures designed for LLMs, ensuring they are introspectable, explainable, and easily synthesized by AI agents.
- Collaborate with internal and external stakeholders to deliver timely LLM-enhanced library releases.
- Develop metrics, tools, and processes to assess the impact and quality of LLM/agentic code generation.
- Prototype tooling and solutions to transform math library APIs into LLM-friendly representations for popular codegen tools.
Requirements
- PhD or MSc in Computational Science, Computer Science, Applied Math, or a related field (or equivalent experience).
- 3+ years of experience.
- Strong knowledge of LLMs, fine-tuning, RL, RAGs, MCP, and agent 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.
- Excellent collaboration, communication, and documentation skills.
Benefits
- Equity
- Benefits