Want to know if this job is worth applying to?
San Jose, California, United States
Onsite
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.
What we need to see:
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
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!
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.You will also be eligible for equity and benefits.
This posting is for an existing vacancy.
NVIDIA uses AI tools in its recruiting processes.
NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.