Senior Software Architect - Deep Learning and Hpc Communications

NVIDIA NVIDIA · Semiconductors · Switzerland +3 · Remote

Senior Software Architect at NVIDIA focused on designing and implementing next-generation data center platforms and scalable communication software to accelerate AI and HPC workloads. The role involves investigating performance bottlenecks, exploring HW/SW co-design, building proofs-of-concept, and performing quantitative modeling for large GPU clusters.

What you'd actually do

  1. Investigate opportunities to improve communication performance by identifying bottlenecks in today's systems.
  2. Design and implement new communication technologies to accelerate AI and HPC workloads.
  3. Explore innovative solutions in HW and SW for our next generation platforms as part of co-design efforts involving GPU, Networking, and SW architects.
  4. Build proofs-of-concept, conduct experiments, and perform quantitive modeling to evaluate and drive new innovations.
  5. Use simulation to explore performance of large GPU clusters (think scales of 100s of 1000s of GPUs)

Skills

Required

  • C/C++ programming
  • Parallel programming models (MPI, SHMEM)
  • Communication runtimes (MPI, NCCL, NVSHMEM, OpenSHMEM, UCX, UCC)
  • Operating systems
  • Computer and system architecture
  • Network architecture
  • Linux

Nice to have

  • CUDA programming
  • NVIDIA GPUs
  • High-performance networks (InfiniBand, RoCE, NVLink)
  • Deep Learning Frameworks (PyTorch, TensorFlow)
  • Deep learning parallelisms
  • HPC applications

What the JD emphasized

  • 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 in fundamentals of network architecture, topology, algorithms, and communication scaling relevant to AI and HPC workloads
  • Strong experience with Linux

Other signals

  • design and implement new communication technologies to accelerate AI and HPC workloads
  • explore innovative solutions in HW and SW for our next generation platforms
  • use simulation to explore performance of large GPU clusters