Accelerate the integration of APIs across the CUDA-X math library software stack into modern LLM and agentic AI workflows. Leading organizations globally use GPU-powered data centers for AI, data analytics, and scientific simulations, driving advancements in fields like LLMs, computer vision, CAE, EDA, and autonomous vehicles. The team develops GPU-accelerated libraries and SDKs essential for these technologies. In this role, you will ensure both human engineers using LLMs and AI agents can build high-performance applications with ease through effective utilization of CUDA-X libraries.
Tasks
- 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.
Requirements
- 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
- Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.
- 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.
Bist du Teil dieses Unternehmens?
Dieses Unternehmensprofil wurde automatisch erstellt. Wenn du für NVIDIA Switzerland AG arbeitest, kannst du das Profil jetzt beanspruchen und verifizieren – kostenlos und in wenigen Minuten.