Senior Software Engineer, NCCL and CUDA - CSP Engagements

About the role

The Senior Software Engineer, NCCL and CUDA specialization role focuses on ML software stack functionality and performance for datacenter products like GB300 and Vera Rubin. This role involves working closely with CSPs to address functional and performance issues in NCCL and CUDA libraries.

Responsibilities

• Engage with 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.
• 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 NCCL.

Requirements

• Experience with parallel programming models and with communication libraries (MPI, NCCL, NVSHMEM) runtime.
• 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.
• 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.

Qualifications

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

Skills

• Knowledge of high-performance networking like InfiniBand, and RoCE.

Benefits

• Base salary range: $184,000 - $287,500 for Level 4, and $224,000 - $356,500 for Level 5.
• Eligibility for equity and benefits.

Pay

Base salary will be determined based on location, experience, and the pay of employees in similar positions.

Schedule

Not specified.

Benefits

• Not specified.

Cookie/Navigation/Equal Opportunity Boilerplate

Not specified.