Senior Math Libraries Engineer – Emulation in AI and HPC

NVIDIA develops GPU accelerated math libraries for AI, scientific, and engineering simulations used in data centers powered by GPUs. The team supports applications in healthcare, NLP, VR, deep learning, and autonomous vehicles.

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

Nice to have:

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

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

Only candidates located in France, Germany, or Poland with remote work are considered.