Senior Math Libraries Engineer - Sparsity in AI

August 18

Apply Now
Logo of NVIDIA

NVIDIA

Artificial Intelligence • Gaming • Automotive

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.

10,000+ employees

Founded 1993

🤖 Artificial Intelligence

🎮 Gaming

📋 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 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 and benefits

Apply Now

Similar Jobs

August 16

Remote AI message quality evaluator focused on tone, empathy, and cultural alignment; supports a US-based team.

August 14

Advanced Korean speakers to train and evaluate AI models at Prolific. Earn about $30/hr per task, with flexible hours and remote work.

🗣️🇰🇷 Korean Required

August 14

Lead analytics engagements using Infor M3 and Birst for Columbus Global.\nRemote across the United States with client-focused delivery.

August 14

Senior Consultant with M3 Analytics and Birst expertise to design, implement, and optimize analytics solutions. Remote role in the United States.

Built by Lior Neu-ner. I'd love to hear your feedback — Get in touch via DM or support@remoterocketship.com