Senior System Software Architect, AI and GPU Networking

at NVIDIA · Industrial · Tel Aviv, Israel +1

This role focuses on architecting and optimizing NVIDIA's GPU Networking offerings for AI workloads, including distributed AI, deep learning, inference, and model serving. It involves co-designing hardware features and leading the architecture and development of new technologies and runtime systems for AI data centers.

What you'd actually do

  1. Enhance NVIDIA's GPU Networking offerings for accelerating AI workloads, such as NVIDIA Dynamo, NVIDIA NIXL and NVIDIA UCX, tailored to the unique requirements of AI workloads.
  2. Co-design hardware features (e.g., in GPUs, DPUs, or interconnects) that accelerate data movement and enable new capabilities for inference and model serving.
  3. Identify and evaluate new technologies, innovations and partner relationships for alignment with our technology roadmap and business value.
  4. Lead architecture and design of new technologies and innovations such as runtime systems, communication libraries, AI-specific technologies.
  5. Lead proof-of-concept development to evaluate and drive such technologies.

Skills

Required

  • M.Sc. or Ph.D. in Computer Science, Electrical or Computer Engineering (or equivalent experience)
  • 5+ years of industry experience in system architecture, AI systems architecture, scaling of AI, Parallelism of AI frameworks, or deep learning training workloads
  • Algorithm design
  • System programming
  • Computer architecture
  • Operating systems
  • Virtualization
  • Networking
  • Storage
  • Performance profiling and optimization techniques
  • Hardware features
  • Programming and software development skills
  • Ability and flexibility to work and communicate effectively in a multi-national, multi-time-zone corporate environment

Nice to have

  • Shown research track record
  • Experience and passion for system architecture, CPU/GPU/memory/storage/networking
  • Stellar communication skills
  • Knowledge in Deep Learning frameworks and AI communication libraries (NCCL, UCX, MPI and equivalents)
  • Deep understanding of Inference and Training workloads and optimizations, like Prefill/Decode, data parallelism, Tensor parallelism, FDSP and others

What the JD emphasized

  • 5+ years of industry experience (or equivalent) in system architecture, AI systems architecture, scaling of AI, Parallelism of AI frameworks, or deep learning training workloads.
  • Deep understanding of performance profiling and optimization techniques, together with defining and using hardware features.
  • Deep understanding of Inference and Training workloads and optimizations, like Prefill/Decode, data parallelism, Tensor parallelism, FDSP and others.

Other signals

  • AI data centers
  • distributed AI and deep learning solutions
  • inference and model serving
  • GPU Networking
Read full job description

NVIDIA has been defining computer graphics, PC gaming, and accelerated computing for more than 25 years. With an outstanding legacy of innovation, driven by phenomenal technology, and extraordinary people, NVIDIA is looking for a strong technical senior architect to join us in shaping the future. Senior Architects are innovators who can translate business needs into workable technology solutions. Their expertise is deep and broad. They are hands on, producing both detailed technical work and high-level architectural designs.

As a Senior architect in the AI Networking Research team, you will explore technological challenges on accelerate networking and building AI data centers. Research new transport functions and semantics for optimizing AI workloads, AI systems communication and accelerations and much more. You will also be leading architectural and development efforts across numerous technological fields, related to the modern AI data center, such as distributed AI and deep learning solutions, data analytics, High Performance Computing (HPC), Software Defined Networking (SDN), virtualization, storage, and more.

What you’ll be doing:

  • Enhance NVIDIA's GPU Networking offerings for accelerating AI workloads, such as NVIDIA Dynamo, NVIDIA NIXL and NVIDIA UCX, tailored to the unique requirements of AI workloads.
  • Co-design hardware features (e.g., in GPUs, DPUs, or interconnects) that accelerate data movement and enable new capabilities for inference and model serving.
  • Identify and evaluate new technologies, innovations and partner relationships for alignment with our technology roadmap and business value.
  • Lead architecture and design of new technologies and innovations such as runtime systems, communication libraries, AI-specific technologies.
  • Lead proof-of-concept development to evaluate and drive such technologies.

What we need to see:

  • Hold a M.Sc. or Ph.D. in Computer Science, Electrical or Computer Engineering from a leading university (or equivalent experience).
  • 5+ years of industry experience (or equivalent) in system architecture, AI systems architecture, scaling of AI, Parallelism of AI frameworks, or deep learning training workloads.
  • Experienced in algorithm design, system programming, computer architecture and operating systems.
  • Experienced in virtualization, networking and storage.
  • Deep understanding of performance profiling and optimization techniques, together with defining and using hardware features.
  • Strong programming and software development skills.
  • Ability and flexibility to work and communicate effectively in a multi-national, multi-time-zone corporate environment.

Ways to stand out from the crowd:

  • Shown research track record.
  • Have experience and passion for system architecture, CPU/GPU/memory/storage/networking.
  • Stellar communication skills.
  • Knowledge in Deep Learning frameworks and AI communication libraries (NCCL, UCX, MPI and equivalents).
  • Deep understanding of Inference and Training workloads and optimizations, like Prefill/Decode, data parallelism, Tensor parallelism, FDSP and others.

NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us. If you're creative and autonomous, we want to hear from you! NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.