Senior Math Libraries Engineer - LLM Integration and Developer Experience

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

Nice To Haves:

  • 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.

What You'll Be Doing:

  • 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.

Perks and 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.

AI Summary ✨

Similar jobs

  • InfoTasks Productivity Agents, DeepMind

    GoogleZurich, Switzerland6 days ago
  • See all jobs in Switzerland →