Distinguished Software Architect - Deep Learning and HPC Communications

NVIDIA; United States; Remote; USD 150,000 - 250,000; Full time

Job summary

NVIDIA is seeking a Distinguished Software Architect to co-design next-generation data center platforms. The role involves researching and developing communication technologies for high-performance computing and deep learning applications, requiring a PhD and extensive experience in relevant fields. Join a leading company at the forefront of AI and HPC innovation.

Qualifications

• 15+ years of relevant experience in academia or industry.
• Expertise in HPC, parallel programming, and GPU architecture.
• Strong programming skills in C or C++ for systems software.

Responsibilities

• Research and design new communication technologies for GPU libraries.
• Propose innovative hardware and software solutions for platforms.
• Drive adoption of new communication technologies across application verticals.

Skills

HPC

Parallel programming models

GPU architecture

CUDA

Deep Learning fundamentals

Performance analysis

C/C++ programming

Education

PhD in Computer Science or related field

Job description

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.

Come work for the team that brought to you NCCL, NVSHMEM & GPUDirect. Our GPU communication libraries are crucial for scaling Deep Learning and HPC applications! We are looking for a Distinguished Software Architect to help co-design our next generation data center platforms. DL and HPC applications have a huge compute demand already and run on scales which go up to tens of thousands of GPUs. The GPUs are connected with high-speed interconnects (eg. NVLink, PCIe) within a node and with high-speed networking (eg. Infiniband, Ethernet) across the nodes. Communication performance between the GPUs has a direct impact on the end-to-end application performance; and the stakes are even higher at huge scales! This is an outstanding opportunity to push the limits on the state-of-the-art and deliver platforms the world has never seen before. Are you ready to contribute to the development of innovative technologies and help realize NVIDIA's vision?

What you will be doing:

• Research new communication technologies (e.g. expand the GPUDirect technology portfolio) and design new features for our communication libraries

• Propose innovative solutions in HW and SW for our next-gen platforms. You will co-design these solutions with the GPU, Networking, and SW architects and ensure seamless integration with the software stacks

• Inspire changes based on quantitative data coming from proof-of-concepts or detailed technical analysis/modeling

• Drive the adoption of new communication technologies across application verticals

• Keep up with the latest DL research and collaborate with diverse teams (internal and external), including DL researchers, and customers

What we need to see:

• PHD in Computer Science, Computer Engineering or related field or strong equivalent experience; 15+ years of relevant experience in academia or the industry

• Expert in following areas: HPC, parallel programming models (MPI, SHMEM), at least one communication runtime (MPI, NCCL, NVSHMEM, OpenSHMEM, UCX, UCC), computer and system architecture, GPU architecture and CUDA

• Deep understanding of various aspects of high performance networking from prior work experience: network technologies (Infiniband, Ethernet), network design, network topologies, network debug and performance analysis

• Strong in at least a few of these areas: ML/DL fundamentals and how they tie to communications, parallel algorithms, fault tolerance and resiliency, competitive assessments, performance analysis and optimizations for parallel applications on large clusters, developing applications using DL Frameworks (PyTorch, TensorFlow)

• Programming fluency with C or C++ for systems software development

• Flexibility to work and communicate effectively across different HW/SW teams and timezones

Ways to stand out from the crowd:

• Industry recognized leader in HPC/DL communications with history of patents, publications and conference talks and keynotes in areas relevant to this role

• Influential role in industry standards (e.g. MPI, OpenSHMEM) and open source software (e.g. PyTorch, UCX, Open MPI)

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 on the planet working for us. If you're creative and autonomous, we want to hear from you!