Senior Math Libraries Engineer - LLM Integration and Developer Experience
Tech Stack
Tag name is followed by "@" symbol and proficiency level value.
About proficiency levels:
- 1-2 — basic awareness. Minimal hands-on experience, and a rudimentary understanding of the technology's purpose;
- 3-6 — daily use. Comfortable and regular usage, capable of handling common tasks and challenges related to the technology;
- 7-9 — you are an expert, you can teach others, you know all the pitfalls and tricks;
- 10 — exceptional knowledge, comprehensive understanding, and adeptness in all aspects of the technology, including advanced problem-solving. Think twice before claiming or demanding such level.
AI @ 4
API @ 4
Agentic AI
CUDA @ 3
Communication @ 4
Computer Vision
GPU
LLM
Mathematics @ 4
RAG @ 4
Reinforcement Learning @ 4
Robotics
- 1-2 — basic awareness. Minimal hands-on experience, and a rudimentary understanding of the technology's purpose;
- 3-6 — daily use. Comfortable and regular usage, capable of handling common tasks and challenges related to the technology;
- 7-9 — you are an expert, you can teach others, you know all the pitfalls and tricks;
- 10 — exceptional knowledge, comprehensive understanding, and adeptness in all aspects of the technology, including advanced problem-solving. Think twice before claiming or demanding such level.
Details
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. Leading organizations globally are using GPU-powered data centers for AI, data analytics, and scientific simulations, driving advancements in fields like LLMs, computer vision, CAE, EDA, and autonomous vehicles. Our team develops the GPU-accelerated libraries and SDKs essential for these technologies.
In this role, you will ensure that human engineers using LLMs and AI agents can build high-performance applications with ease through effective utilization of CUDA-X libraries. The role requires a blend of technical, developer experience, and communication skills.
Responsibilities
- Drive the architecture and implementation of math library 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 and agentic code generation.
- Prototype tooling and solutions to transform math library APIs into LLM-friendly representations across popular code-generation tools.
Requirements
- PhD or master's degree in Computational Science, Computer Science, Applied Mathematics, or a related science or engineering field, preferred; equivalent experience may be considered.
- At least 3 years of experience.
- Robust knowledge of LLMs, fine-tuning, reinforcement learning, retrieval-augmented generation (RAG), Model Context Protocol (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 with minimal direction.
- Excellent collaboration, communication, and documentation skills.
Preferred Qualifications
- Experience in AI-assisted software engineering, code generation, or programming language design.
- Familiarity with CUDA-X math library APIs, including cuBLAS, cuFFT, cuSOLVER, and cuSPARSE.
Company
NVIDIA develops GPU-accelerated technologies, libraries, and SDKs for AI, data analytics, scientific simulations, computer graphics, robotics, and autonomous vehicles.
Compensation
For Poland, the base salary range is 221,250 PLN–383,500 PLN for Level 3 and 292,500 PLN–507,000 PLN for Level 4. Base salary is determined based on location, experience, and the pay of employees in similar positions.