Senior Math Libraries Engineer – AI and HPC
$184k - $357k • US, CA, Santa Clara
Posted 14d ago
Job Location
US, CA, Santa Clara
Tech Stack
Remote Work Policy
On-site
Categories
Machine Learning Engineer
About the job
NVIDIA's Math Libraries team is seeking a senior engineer to contribute to kernel generation for AI and HPC, focusing on matrix operations, JITing, and fusions. This role involves developing GPU-accelerated mathematical libraries that power advancements in AI, scientific simulations, and data analytics across various industries. You will be instrumental in enabling leading organizations to revolutionize fields like healthcare, NLP, VR, and autonomous vehicles.
Responsibilities
- Design and implement high-performance numerical dense linear algebra software on GPUs.
- Lead projects involving multiple engineers and teams.
- Provide technical leadership and mentorship to library engineers.
- Collaborate with product management and customers to define feature and performance requirements.
- Identify opportunities to improve library performance and reduce code maintenance through re-architecting.
- Find and explain complex solutions, exercise leadership, and coordinate cross-team efforts.
Requirements
- PhD, Master’s, or Bachelor’s degree in Computer Science, Applied Math, or a related field (or equivalent experience).
- 8+ years of experience in designing, developing, testing, maintaining, and optimizing HPC software using C++.
- Strong fundamentals in kernel generation and composable library design for linear algebra.
- Demonstrated leadership skills in driving software development projects.
- Strong collaboration, communication, and documentation skills.
- Experience with kernel generation and JIT (Just-In-Time) compilation.
Benefits
- Competitive salary package
- Equity
- Benefits