Our AI Networking team is looking to hire a student interested in researching and developing solutions for the most pressing networking challenges of the world's fastest AI and HPC supercomputers. As a member of our team, you will be responsible for investigating, crafting, and implementing new protocols and algorithms that enable the best performance possible on NVIDIA networked supercomputers. This role offers you an excellent opportunity to deliver production-grade solutions, get hands-on with ground-breaking technology, and work closely with top technical leaders in shaping and innovating next-generation networking technologies.

What you'll be doing:

The team is responsible for developing high-performance communication frameworks running in the production of the largest AI clusters. The work environment is dynamic and adventurous; we are innovating and inventing software products at the forefront of technology in terms of performance, scalability, and features. In addition, our team works closely with networking chip design teams in co-designing new hardware features and software APIs.

What we need to see:

• Student for M.Sc. or Ph.D. in Computer Science, Software Engineering, Electrical Engineering, etc.
• Experience in ML/AI workloads, particularly in distributed training, including knowledge in Pytorch or equivalent
• Experience in Programming in C/C++ and Python, together with Linux environment and tools
• Problem-Solving Approach: Ability to break down complex problems into actionable components.
• Knowledge of computer architecture and operating systems

Ways to stand out from the crowd:

• Knowledge of communication libraries such as NCCL, UCX, and UCC.
• Knowledge of Networking Protocols such as Ethernet, InfiniBand, IP, TCP
• Previous work experience as a software developer
• Open Source Software Contributor

We are at the beginning of the next era, the AI computing era, ignited by GPU deep learning, which has shown to be deeply effective at solving some of the most complex problems in everyday life. NVIDIA is driving these efforts worldwide. Are you creative and driven? Do you love a challenge? If so, we want to hear from you.

We are committed to encouraging a diverse work environment and are 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. We will ensure that individuals with disabilities are provided reasonable accommodation to participate in the job application or interview process, perform essential job functions, and receive other benefits and privileges of employment. Please contact us to request accommodation.