[Remote] Senior Math Libraries Engineer - LLM Integration and Developer Experience

Note: The job is a remote job and is open to candidates in USA. NVIDIA is a leading technology company known for its innovations in GPU computing and AI. They are seeking a Senior Math Libraries Engineer to enhance the integration of APIs within the CUDA-X math library software stack for LLM and AI workflows, ensuring high-performance application development for engineers and AI agents alike. 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 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 Skills • PhD or MSc's degree in Computational Science, Computer Science, Applied Math, or 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 • 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 Benefits • Equity • Benefits Company Overview • NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI. It was founded in 1993, and is headquartered in Santa Clara, California, USA, with a workforce of 10001+ employees. Its website is Company H1B Sponsorship • NVIDIA has a track record of offering H1B sponsorships, with 1877 in 2025, 1355 in 2024, 976 in 2023, 835 in 2022, 601 in 2021, 529 in 2020. Please note that this does not guarantee sponsorship for this specific role. Apply tot his job