Senior Math Libraries Engineer – AI and HPC
Join to apply for the Senior Math Libraries Engineer – AI and HPC role at NVIDIA
Senior Math Libraries Engineer – AI and HPC
Join to apply for the Senior Math Libraries Engineer – AI and HPC role at NVIDIA
Get AI-powered advice on this job and more exclusive features.
Direct message the job poster from NVIDIA
Passionate About AI & Machine Learning - NVIDIA
NVIDIA Math Libraries team is looking for a senior engineer to join our development efforts in the area of kernel generation for AI and HPC, specifically targeting matrix operations, JITing and fusions. Around the world, leading commercial and academic organizations are revolutionizing AI, scientific and engineering simulations, and data analytics, using data centers powered by GPUs. Applications of these technologies are in healthcare, NLP, VR, deep learning, autonomous vehicles and countless others. Did you know our team develops the GPU accelerated mathematical libraries that makes all of this possible?
What You Will Be Doing
- Scoping, designing, and implementing high quality and performance numerical dense linear algebra software on GPUs.
- Owning the execution of projects involving multiple engineers and sometimes teams.
- Providing technical leadership and feedback to library engineers working with you on projects and sometimes mentor interns.
- Working closely with product management and other internal and external customers to understand feature and performance requirements and contribute to the technical roadmaps of libraries.
- Finding opportunities to improve library performance and reduce code maintenance overhead through re-architecting.
- To 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 goals.
- PhD, 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 algebra.
- Leadership skills in driving software development projects.
- Strong collaboration, communication, and documentation habits.
- Kernel generation. JIT focus/experience desired
- Experience with parallel programming, ideally using CUDA, MPI, OpenMP, OpenACC, pthreads.
- Good understanding of Machine Learning and Deep Learning technologies as well as knowledge of GPU (preferred) or CPU hardware architecture.
- Experience with low level programming using assembly for performance optimization and operator fusion is a huge plus.
- Experience with agile software development practices using project management tools such as JIRA.
- A scripting language, preferably Python.
The base salary range is 184,000 USD - 356,500 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.
You will also be eligible for equity and benefits . NVIDIA accepts applications on an ongoing basis.
NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.
JR1998721
Seniority level
Seniority level
Mid-Senior level
Employment type
Employment type
Full-time
Job function
Industries
Computer Hardware Manufacturing, Software Development, and Computers and Electronics Manufacturing
Referrals increase your chances of interviewing at NVIDIA by 2x
Get notified about new Senior Software Engineer jobs in California, United States .
Los Angeles, CA $160,000.00-$190,000.00 3 weeks ago
Founding Team Senior Software Engineer Manager for a Social Travel Platform targeting Digital Nomads
San Francisco, CA $150,000.00-$200,000.00 4 months ago
California, United States $171,000.00-$241,700.00 2 weeks ago
Sr. Software Engineer - Marketing (SFMC)
Beverly Hills, CA $128,000.00-$160,000.00 1 week ago
Software Architect / Digital Dev & Soln Arch / Full-time / Days
Senior Developer Relations Manager, CSP Hyperscale
California, United States $224,000.00-$425,500.00 2 days ago
Senior Engineer - Warehouse Management System
Brisbane, CA $140,000.00-$180,000.00 1 week ago
Senior Systems Engineer (R&D and Product Development) - Remote
San Francisco, CA $95,000.00-$230,000.00 4 months ago
Software Engineering Leader - SLS, 3D Imaging, & Point Cloud
San Francisco, CA $150,000.00-$250,000.00 2 days ago
Senior Full Stack Software Engineer (Billing)
Senior Full Stack Software Engineer (Vault)
Santa Barbara, CA $140,000.00-$200,000.00 1 month ago
Los Angeles, CA $140,000.00-$170,000.00 1 month ago
San Francisco, CA $150,000.00-$207,000.00 5 months ago
Senior Software Engineer, Backend - Fintech
United States $100,000.00-$700,000.00 3 weeks ago
San Francisco, CA $154,000.00-$210,000.00 4 weeks ago
California, United States $164,500.00-$247,600.00 1 week ago
San Francisco, CA $180,000.00-$220,000.00 8 months ago
Irvine, CA $160,000.00-$180,000.00 3 days ago
Senior Software Engineer, Game Experience
United States $196,000.00-$220,500.00 1 week ago
We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.
#J-18808-Ljbffr