Senior Math Libraries Engineer – Emulation in AI and HPC

Senior Math Libraries Engineer – Emulation in AI and HPC

Join NVIDIA as a Senior Math Libraries Engineer focusing on emulation of math operations across different precisions for AI and HPC kernels. Our team develops GPU‑accelerated math libraries that power healthcare, NLP, VR, deep learning, autonomous vehicles, and more.

What You Will Be Doing

• Scoping, designing, and implementing high‑quality, high‑performance numerical dense linear algebra software on GPUs.
• Providing technical leadership and feedback to library engineers and mentoring interns.
• Working closely with product management and internal/external customers to understand feature and performance requirements and help define library roadmaps.
• Finding opportunities to improve library performance and reduce code maintenance overhead through re‑architecting.

What We Need To See

• PhD or Master’s degree in Computer Science, Applied Math, or a related science or engineering field (or equivalent experience).
• 5+ years of experience designing, developing, testing, maintaining, and performance‑optimizing production software using CUDA and C++.
• Good knowledge of GPU (preferred) or CPU hardware architecture.
• Strong fundamentals in finite‑precision arithmetics and numerical methods for linear algebra.
• Excellent teamwork, communication, and documentation habits.

Ways To Stand Out From The Crowd

• Experience with CUTLASS or low‑level programming such as assembly for performance optimization.
• Proficiency in a scripting language, preferably Python.
• Experience working in a globally distributed team.

Competitive salaries and a generous benefits package are offered. Join us on the forefront of technological advancement.

Seniority level: Mid‑Senior level

Employment type: Full‑time

Job function: Computer Hardware Manufacturing, Software Development, and Computers and Electronics Manufacturing

#J-18808-Ljbffr