Senior System Software Engineer, NCCL - Partner Enablement

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. Our work opens up new universes to explore, enables amazing creativity and discovery, and powers what were once science fiction inventions from artificial intelligence to autonomous cars.

We are the GPU Communications Libraries and Networking team at NVIDIA. We deliver communication runtimes like NCCL and NVSHMEM for Deep Learning and HPC applications. We are looking for a motivated Partner Enablement Engineer to guide our key partners and customers with NCCL. Most DL/HPC applications run on large clusters with high-speed networking (Infiniband, RoCE, Ethernet). This is an outstanding opportunity to get an end to end understanding of the AI networking stack.

As a Senior System Software Engineer, NCCL - Partner Enablement, you will:

• Engage with partners and customers to root cause functional and performance issues reported with NCCL
• Conduct performance characterization and analysis of NCCL and DL applications on groundbreaking GPU clusters
• Develop tools and automation to isolate issues on new systems and platforms, including cloud platforms (Azure, AWS, GCP, etc.)
• Guide customers and support teams on HPC knowledge and standard methodologies for running applications on multi-node clusters
• Document and conduct trainings/webinars for NCCL
• Engage with internal teams in different time zones on networking, GPUs, storage, infrastructure and support

Requirements:

• B.S./M.S. degree in CS/CE or equivalent experience with 5+ years of relevant experience
• Experience with parallel programming and at least one communication runtime (MPI, NCCL, UCX, NVSHMEM)
• Excellent C/C++ programming skills, including debugging, profiling, code optimization, performance analysis, and test design
• Experience working with engineering or academic research community supporting HPC or AI
• Practical experience with high performance networking: Infiniband/RoCE/Ethernet networks, RDMA, topologies, congestion control
• Expert in Linux fundamentals and a scripting language, preferably Python
• Familiar with containers, cloud provisioning and scheduling tools (Docker, Docker Swarm, Kubernetes, SLURM, Ansible)
• Adaptability and passion to learn new areas and tools
• Flexibility to work and communicate effectively across different teams and timezones

Join NVIDIA and be part of a team that's shaping the future of AI and high-performance computing!