Senior Math Libraries Engineer – Sparsity in AI

🕒 April 10

🏄 California – Remote

🏄 California – Remote

💵 $184k - $356.5k / year

⏰ Full Time

🟠 Senior

🤖 Artificial Intelligence

🦅 H1B Visa Sponsor

This company has sponsored H1B visas in the past.

Apply Now

Find Similar Remote Jobs

Set Up Email Alert For New Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Upload Resume

NVIDIA

WebsiteLinkedInAll Job Openings

10,000+ employees

Founded 1993

🏥 Healthcare

🏭 Manufacturing

🤖 Artificial Intelligence

Healthcare • Manufacturing • Artificial Intelligence

NVIDIA is a leading technology company specializing in accelerated computing and artificial intelligence. NVIDIA pioneers advancements in graphical processing units (GPUs), cloud computing, data centers, and virtual reality, with a focus on gaming, automotive, healthcare, and robotics industries. The company's innovations, such as NVIDIA Omniverse, transform traditional digital processes by enabling high-fidelity simulations and rendering tasks. Their applications span various industries, from autonomous vehicles using NVIDIA DRIVE to healthcare solutions with NVIDIA Clara, and AI-driven analytics and workflows.

📋 Description

• Design and develop a C++-based system to simplify and accelerate computing for unstructured sparsity in DL and HPC on NVIDIA GPUs • Enable the system in languages and frameworks that are more commonly used in DL, such as Python and PyTorch • Evaluate and improve the performance of the system on real-life applications • Realize opportunities to improve library quality, performance and maintainability by writing effective and well-tested code for production use • Work closely with product management and other internal and external partners to understand feature and performance requirements and contribute to technical roadmaps

🎯 Requirements

• BS, MS or PhD degree in Computer Science, Applied Math, or related field (or equivalent experience) • 6+ years of overall experience in developing, debugging and optimizing high-performance software using C++ and parallel programming; ideally for sparse linear algebra applications and using CUDA, MPI, OpenMP, or equivalent technologies • Experience with domain-specific language design and compiler optimizations, in particular sparse compilers (MLIR or TACO) • Excellent C++, Python, and CUDA programming skills • Strong collaboration, communication, and documentation habits and ideally experience with working in a globally distributed organization

🏖️ Benefits

• equity • benefits