Senior Software Engineer, NCCL and CUDA - CSP Engagements

NVIDIA Ltd.
Santa Clara, United States of America
yesterday

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Working hours
Regular working hours
Languages
English
Experience level
Senior
Compensation
$ 168K

Job location

Santa Clara, United States of America

Tech stack

C++
Profiling
Nvidia CUDA
Computer Programming
Computer Engineering
Data Centers
ETL
Software Debugging
InfiniBand
PCI Express
Performance Tuning
Software Architecture
Cloud Services
Ansible
Systems Architecture
System Software
Deep Learning
Parallel Computation
Kubernetes
Information Technology
Bare Metal
Slurm
Docker

Job description

NVIDIA is seeking a Senior Software Engineer, NCCL and CUDA specialization to join our Cloud Service Provider (CSP)Engagements team, focusing on ML software stack functionality and performance for datacenter products such as GB300 and Vera Rubin. This role involves working with the customers to understand the functional and performance issues in the libraries layer for deployment at scale. The role combines deep technical expertise in workloads, NCCL and CUDA libraries, frameworks, and system software interaction to solve the customer issues and bring the next generation of innovation.

What you will be doing:

  • Engage with our CSPs to root cause functional and performance issues in NCCL and CUDA libraries.
  • Analyze and improve multi-GPU workloads performance through profiling, benchmarking, and tuning.
  • Understand and solve NCCL and NVSHMEM data movement issues in multi-node clusters.
  • Understand and solve CUDA porting issues for customer workloads.
  • Apply datacenter-specific scheduling and topologies for optimal performance
  • Debug and resolve complex issues related to GPU computation, memory, and transports.
  • Collaborate with customers to understand their workload integration specific challenges to NCCL and CUDIA libraries and suggest tailored solutions aligned with the NVIDIA ecosystem.
  • Collaborate with AE, FAE, and solution architects to deliver integrated customer solutions and technical documentation.
  • Collaborate with internal teams to help customers use the latest advancements in CUDA and in NCCL.

Requirements

  • Experience with parallel programming models and with communication libraries (MPI, NCCL, NVSHMEM) run time.
  • Experience with performance optimization and profiling tools (e.g., Nsight, nvprof)
  • Excellent C/C++ programming and debugging skills, with experience in CUDA development.
  • Good exposure to PCIe and NVLINK.
  • Deep understanding of operating systems and data-center system architecture.
  • Knowledge of high-performance networking like InfiniBand, and RoCE.
  • Proficient understanding of compute, networking and cloud deployment, specifically on bare-metal and VMs.
  • BS or MS in Computer Engineering, Computer Science, or related field (or equivalent experience).
  • Familiarity with containers, cloud provisioning and scheduling tools such as Docker, Kubernetes, SLURM, and Ansible.
  • 8+ years of system software validation experience.
  • Ability to communicate effectively and collaborate with partner and customer teams.

Ways to stand out from the crowd:

  • Strong software architecture experience.
  • Experience with deep learning workloads training and inferencing
  • Experience conducting performance benchmarking and developing tooling on HPC clusters

Benefits & conditions

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., $90,000.00 - $157,500.00 per year

About the company

NVIDIA is leading the way in groundbreaking developments in Artificial Intelligence, High-Performance Computing and Visualization. The GPU, our invention, serves as the visual cortex of modern computers and is at the heart of our products and services. We have some of the most forward-thinking and hard-working people on the planet working for us. If you're creative, hard-working and self-motivated, we want to hear from you!

Apply for this position