Used Tools & Technologies
Not specified
Required Skills & Competences ?
Software Development @ 4 Python @ 4 Machine Learning @ 7 Leadership @ 4 Communication @ 7 Parallel Programming @ 4 Performance Optimization @ 7 Jira @ 4 Product Management @ 4 Technical Leadership @ 4 Project Management @ 4 NLP @ 4 Agile @ 4 CUDA @ 4 GPU @ 4Details
NVIDIA Math Libraries team is looking for a senior engineer to join development efforts in kernel generation for AI and HPC, specifically targeting matrix operations, JITing and fusions. The team develops GPU-accelerated mathematical libraries used in AI, scientific and engineering simulations, and data analytics across industries such as healthcare, NLP, VR, deep learning, and autonomous vehicles.
Responsibilities
- Scope, design, and implement high-quality, high-performance numerical dense linear algebra software on GPUs.
- Own execution of projects involving multiple engineers and sometimes teams.
- Provide technical leadership and feedback to library engineers and sometimes mentor interns.
- Work closely with product management and other internal and external customers to understand feature and performance requirements and contribute to technical roadmaps.
- Find opportunities to improve library performance and reduce code maintenance through re-architecting.
- Explain complex solutions, exercise leadership, and coordinate with multiple teams to achieve goals.
Requirements
- PhD, Master’s, or Bachelor’s degree in Computer Science, Applied Math, or related science or engineering field (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 and JIT focus/experience desired.
Ways to stand out / Preferred qualifications
- Experience with parallel programming, ideally using CUDA, MPI, OpenMP, OpenACC, pthreads.
- Good understanding of Machine Learning and Deep Learning technologies and knowledge of GPU (preferred) or CPU hardware architecture.
- Experience with low-level programming (assembly) for performance optimization and operator fusion is a huge plus.
- Experience with agile software development practices and project management tools such as JIRA.
- Experience with a scripting language, preferably Python.
Compensation and Benefits
Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5. You will also be eligible for equity and benefits.
Other information
- Applications for this job will be accepted at least until July 29, 2025.
- NVIDIA is an equal opportunity employer and is committed to fostering a diverse work environment.