Senior Math Libraries Engineer – LLM Integration, Developer Experience

This is a fully remote position, open to applicants in California, +4 more states.

📋 Description

• Lead the design and execution of mathematical library APIs and documentation frameworks that prioritize LLM capabilities: introspectable, explainable, and easily interpretable by AI agents.

• Collaborate with both internal and external stakeholders to ensure timely releases of LLM-enhanced library updates.

• Develop metrics, tools, and processes to assess the impact and quality of LLM/agentic code generation.

• Create prototypes and solutions to adapt the APIs of math libraries into LLM-compatible formats across widely-used code generation tools.

⛳️ Requirements

• A PhD or MSc degree in Computational Science, Computer Science, Applied Mathematics, or a related scientific or engineering discipline is preferred (or equivalent experience).

• A minimum of 3 years of relevant experience.

• Strong understanding of LLMs, fine-tuning, reinforcement learning, constructing RAGs, MCP, and developing agenting tools.

• Demonstrated experience in designing clear, composable APIs and producing high-quality, well-documented code within complex technical fields.

• Capacity to manage multiple projects and operate independently with minimal supervision.

• Exceptional collaboration, communication, and documentation skills.

• Previous experience in AI-assisted software engineering, code generation, or programming language design.

• Familiarity with CUDA-X math library APIs, including cuBLAS, cuFFT, cuSOLVER, cuSPARSE, etc.

🏝️ Benefits

• Equity

• Benefits