AI Networking Software Developer

Nvidia • Sunnyvale, CA

432d • $120,000 - $276,000

About The Position

NVIDIA is seeking an AI Networking Software Developer to join their AI Networking SW development and codesign team. This role focuses on designing and implementing new protocols and algorithms to optimize performance on NVIDIA's networked supercomputers and data centers. The developer will work on production-grade solutions, engage with innovative technology, and collaborate with technical leaders to tackle significant challenges in machine learning and cloud computing.

Requirements

• Master's degree in computer science or software engineering, or equivalent experience.
• 2+ years of professional experience in high performance computing or a closely related field.
• Professional experience programming in C/C++.
• Experience working in a Linux environment and with Linux tools.
• Deep knowledge of Networking Protocols, specifically InfiniBand and Ethernet.
• Deep knowledge in computer architecture and operating systems.
• Experience in performance optimizations.

Nice To Haves

• PhD in Computer Science, Electrical Engineering, Mathematics, or Physics.
• Knowledge in MPI, PyTorch, JAX, and other HPC and AI frameworks.
• Knowledge in RDMA technology.
• Hands-on experience developing and deploying production-grade software.

Responsibilities

• Developing AI networking communication frameworks and applications for production use on supercomputers and data centers.
• Innovating and inventing software products that enhance performance, scalability, and features.
• Collaborating with networking teams to co-design new hardware features and software APIs.
• Working in a dynamic and challenging environment.

Benefits

• Equity options
• Comprehensive health benefits
• Flexible work arrangements
• Diversity and inclusion programs

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Industry: Computer and Electronic Product Manufacturing
• Education Level: Master's degree