Software Engineer - CUDA-X Math Library Integration

We are seeking a skilled software engineer to accelerate the integration of APIs across the CUDA-X math library software stack into modern LLM and agentic AI workflows. You will be responsible for ensuring that both human engineers using LLMs and AI agents build high-performance applications with ease through effective utilization of CUDA-X libraries.

Responsibilities:

• Drive the architecture and implementation of math libraries APIs and documentation structures designed to be LLM-first: introspectable, explainable, and easily synthesized by AI agents.
• Work with internal and external 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 help transform math libraries' APIs into LLM-friendly representations across popular codegen tools.

Requirements:

• PhD or MSc's degree in Computational Science, Computer Science, Applied Math, or a related science or engineering field of study is preferred (or equivalent experience).
• 3+ years of experience.
• Robust knowledge of LLMs, finetuning, RL, building RAGs, MCP, and building agenting 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 with minimal direction.
• Excellent collaboration, communication, and documentation habits.

Preferred Qualifications:

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

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 152,000 USD - 241,500 USD for Level 3, and 184,000 USD - 287,500 USD for Level 4. You will also be eligible for equity and benefits.