About the Role
NVIDIA is seeking an AI Networking Software Developer to join their AI Networking SW development and codesign team. This role involves working with cutting-edge NVIDIA hardware and technologies, performing full-stack benchmarking for AI training/inference on data center-scale systems, and developing in-house tools while adopting community practices. The goal is to ensure large-scale systems deliver expected performance by identifying bottlenecks and driving improvements.
Responsibilities
- Developing AI networking communication frameworks and applications for production on large-scale supercomputers and data centers.
- Developing production tools and benchmarks utilized by internal and external NVIDIA teams.
- Enabling new AI models within the benchmarking infrastructure.
- Delivering insights through end-to-end analysis of large-scale workloads across hardware and software stacks.
- Designing and implementing automation systems, including large-scale parameter search for optimal configurations.
- Collaborating with networking and hardware teams for co-design of new features and software interfaces.
Requirements
- B.Sc., M.Sc degree in Computer Science / Software engineering or equivalent experience.
- 5+ years of experience.
- Professional Python development experience.
- Experience building maintainable, long-lived tools with low maintenance burden.
- Solid Linux expertise and extensive command-line environment experience.
- Ability to work across a broad and evolving stack.
- Strong drive to learn, from hardware and networking up to large-scale AI systems.
Skills
- Python
- Linux
- Command-line environments
- Modern AI ecosystem (PyTorch, LLMs, inference, training)
- Cluster orchestration systems (Slurm, Kubernetes)
- MPI
- HPC
- InfiniBand
- Ethernet
- Networking
- Performance optimizations
Location
- Remote
Work Type
- Full-time
Experience Level
- 5+ years of experience
Education Level
- B.Sc., M.Sc degree in Computer Science / Software engineering or equivalent experience
Salary/Compensations
- 152,000 USD - 241,500 USD for Level 3
- 184,000 USD - 287,500 USD for Level 4
Benefits
- Equity
- Benefits
About the Company
- For two decades, NVIDIA has pioneered visual computing and computer graphics, inventing the GPU. The field has expanded to encompass video games, movie production, product design, medical diagnosis, and scientific research. NVIDIA is now at the beginning of the AI computing era, driven by GPU deep learning. NVIDIA is considered one of the technology world's most desirable companies.
- NVIDIA pioneered accelerated computing. Today, our AI infrastructure powers global intelligence, transforming every industry.
Equal Opportunity
- NVIDIA is committed to fostering an inclusive 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.
