[Remote] Senior Math Libraries Engineer – AI and HPC

Note: The job is a remote job and is open to candidates in USA. NVIDIA is a leading technology company specializing in AI and HPC solutions. They are seeking a Senior Math Libraries Engineer to develop high-quality numerical dense linear algebra software on GPUs, while providing technical leadership and collaborating with product management and customers to enhance library performance.ResponsibilitiesScoping, designing, and implementing high quality and performance numerical dense linear algebra software on GPUsOwning the execution of projects involving multiple engineers and sometimes teamsProviding technical leadership and feedback to library engineers working with you on projects and sometimes mentor internsWorking closely with product management and other internal and external customers to understand feature and performance requirements and contribute to the technical roadmaps of librariesFinding opportunities to improve library performance and reduce code maintenance overhead through re-architectingTo be successful in your responsibilities which are by nature sophisticated, you will need to find and explain complex solutions, exercise leadership, and coordinate with multiple teams to work towards your goalsSkillsPhD, Master's, or Bachelor's degree in Computer Science, Applied Math, or related science or engineering field of study (or equivalent experience)8+ years of experience in designing, developing, testing, maintenance, and performance optimization of HPC software using C++Strong fundamentals in kernel generation and composable library design for linear algebraLeadership skills in driving software development projectsStrong collaboration, communication, and documentation habitsKernel generation. JIT focus/experience desiredExperience with parallel programming, ideally using CUDA, MPI, OpenMP, OpenACC, pthreadsGood understanding of Machine Learning and Deep Learning technologies as well as knowledge of GPU (preferred) or CPU hardware architectureExperience with low level programming using assembly for performance optimization and operator fusion is a huge plusExperience with agile software development practices using project management tools such as JIRAA scripting language, preferably PythonBenefitsEquityBenefitsCompany OverviewNVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI. It was founded in 1993, and is headquartered in Santa Clara, California, USA, with a workforce of 10001+ employees. Its website is https://www.nvidia.com.Company H1B SponsorshipNVIDIA has a track record of offering H1B sponsorships, with 448 in 2026, 1872 in 2025, 1354 in 2024, 976 in 2023, 835 in 2022, 601 in 2021, 529 in 2020. Please note that this does not guarantee sponsorship for this specific role.