Job Summary:
NVIDIA is seeking a Senior Software Engineer with NCCL and CUDA specialization to join their Cloud Service Provider Engagements team, focusing on enhancing ML software stack performance for datacenter products. The role involves collaborating with customers to address functional and performance issues in libraries, applying technical expertise to optimize workloads, and delivering integrated solutions aligned with NVIDIA's ecosystem.
Responsibilities:
โข 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.
Qualifications:
Required:
โข 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.
Preferred:
โข Strong software architecture experience.
โข Experience with deep learning workloads training and inferencing
โข Experience conducting performance benchmarking and developing tooling on HPC clusters
Company:
NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI. Founded in 1993, the company is headquartered in Santa Clara, USA, with a team of 10001+ employees. The company is currently Late Stage.