We are seeking software engineers to design and develop libraries and tools that enhance computing for unstructured sparsity in deep learning and high-performance computing. The team focuses on creating GPU-accelerated libraries and SDKs that support various applications, including AI and scientific simulations. Ideal candidates will have experience in accelerated computing software and a passion for advancing technology in this field. This role offers the opportunity to work with cutting-edge technologies and contribute to significant advancements in AI and data analytics.

• BS, MS or PhD degree in Computer Science, Applied Math, or related field or equivalent experience
• 6+ years of experience in developing, debugging, and optimizing high-performance software using C++ and parallel programming
• Experience with domain-specific language design and compiler optimizations, particularly sparse compilers like MLIR or TACO
• Excellent programming skills in C++, Python, and CUDA
• Strong collaboration, communication, and documentation habits

• Design and develop a C++-based system to simplify and accelerate computing for unstructured sparsity in deep learning and high-performance computing on NVIDIA GPUs
• Enable the system in languages and frameworks commonly used in deep learning, such as Python and PyTorch
• Evaluate and improve the performance of the system on real-life applications
• Write effective and well-tested code for production use to enhance library quality, performance, and maintainability
• Collaborate with product management and partners to understand feature and performance requirements and contribute to technical roadmaps

• Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.