LLM-First Math Libraries Engineer & DevEx Specialist

NVIDIA; Lavamünd; On-site; EUR 90,000 - 150,000; Full time

Job summary

NVIDIA in Austria is seeking a passionate engineer to accelerate API integration across CUDA-X math libraries for LLM and AI workflows. You will design LLM-first APIs, ensure clear documentation, and collaborate with researchers and engineers to deliver high-performance, AI-ready math libraries.

The role emphasizes independent work, strong communication, and a track record in API design and GPU-accelerated software.

Qualifications

• PhD or MSc in Computational Science, Computer Science, Applied Math, or related field.
• 3+ years of experience in relevant roles.
• Robust knowledge of LLMs, finetuning, RL, RAGs, and agent tooling.
• Proven API design experience with high-quality, documented code.
• Ability to prioritize multiple projects and work independently.
• Excellent collaboration and communication habits.

Responsibilities

• Drive architecture and implementation of math library APIs designed for LLM-first use.
• Collaborate with stakeholders to deliver timely LLM-enhanced library releases.
• Build metrics and tooling to measure impact of LLM/agentic code generation.
• Prototype tooling to transform APIs into AI-friendly representations across codegen tools.

Skills

LLMs

API design

Agent tooling

Documentation

Education

PhD or MSc in Computational Science or related

Tools

cuBLAS

cuFFT

cuSOLVER

cuSPARSE

Job description

NVIDIA in Austria is seeking a passionate engineer to accelerate API integration across CUDA-X math libraries for LLM and AI workflows. You will design LLM-first APIs, ensure clear documentation, and collaborate with researchers and engineers to deliver high-performance, AI-ready math libraries.

The role emphasizes independent work, strong communication, and a track record in API design and GPU-accelerated software.