Senior Software Architect - Deep Learning and HPC Communications

Senior Software Architect – Deep Learning and HPC Communications

NVIDIA is leading groundbreaking developments in Artificial Intelligence, High Performance Computing and Visualization. Our GPU—our invention—serves as the visual cortex of modern computers and is at the heart of our products and services. Join the team that developed NCCL, NVSHMEM & GPUDirect and help co‑design next‑gen data center platforms and scalable communications software to accelerate AI and HPC workloads.

What You Will Be Doing

• Investigate opportunities to improve communication performance by identifying bottlenecks in today’s systems.
• Design and implement new communication technologies to accelerate AI and HPC workloads.
• Explore innovative solutions in HW and SW for our next‑generation platforms with GPU, networking, and software architects.
• Build proofs‑of‑concept, conduct experiments, and perform quantitative modeling to evaluate and drive new innovations.
• Use simulation to explore performance of large GPU clusters (scales of hundreds to thousands of GPUs).

What We Need To See

• M.S./Ph.D. degree in CS/CE or equivalent experience.
• 5+ years of relevant experience.
• Excellent C/C++ programming and debugging skills.
• Experience with parallel programming models (MPI, SHMEM) and at least one communication runtime (MPI, NCCL, NVSHMEM, OpenSHMEM, UCX, UCC).
• Deep understanding of operating systems, computer and system architecture.
• Solid fundamentals of network architecture, topology, algorithms, and communication scaling relevant to AI and HPC workloads.
• Strong experience with Linux.
• Ability and flexibility to work and communicate effectively in a multi‑national, multi‑time‑zone corporate environment.

Ways To Stand Out From The Crowd

• Expertise in related technology and passion for what you do; experience with CUDA programming and NVIDIA GPUs; knowledge of high‑performance networks like InfiniBand, RoCE, NVLink, etc.
• Experience with deep learning frameworks such as PyTorch, TensorFlow, etc.; knowledge of deep learning parallelisms and mapping to the communication subsystem; experience with HPC applications.
• Strong collaborative and interpersonal skills and a proven track record of effectively guiding and influencing within a dynamic and multi‑functional environment.

As an equal opportunity employer, we are committed to fostering a supportive and empowering workplace for all. NVIDIA offers highly competitive salaries, an extensive benefits package, and a work environment that promotes diversity, inclusion, and flexibility.

Job ID: JR2008062

Employment type: Full‑time. Seniority level: Mid‑Senior.