Senior Software Developer, AI Networking

NVIDIA; California (MO); On-site; USD 184,000 - 287,500; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is seeking an AI Networking Software Developer to join our AI Networking SW development and codesign team. You will work on production-ready AI networking frameworks and benchmarks on some of the world’s largest data centers and supercomputers.

You will develop tools that scale across teams, enable new AI models within benchmarking pipelines, and collaborate with networking and hardware teams to optimize performance, automate experiments, and improve end-to-end analytics for complex

Qualifications

• B.Sc./M.Sc in CS or software engineering, or equivalent experience.
• At least 5+ years of professional software development experience.
• Strong Python, Linux and command-line proficiency.
• Ability to learn across a broad hardware-software stack.

Responsibilities

• Develop AI networking communication frameworks and production-ready applications for large-scale data centers.
• Create production tools and benchmarks used by multiple NVIDIA teams.
• Enable new AI models within benchmarking infrastructure and analyze large-scale workloads across hardware and software stacks.
• Design and implement automation systems, including large-scale parameter search for optimal configurations.
• Collaborate with networking and hardware teams to co-design features and interfaces in a fast-paced environment.

Skills

Python

Linux

5+ years experience

Command-line tools

Software development

Education

B.Sc./M.Sc in CS or software engineering

Tools

Slurm

Kubernetes

MPI

InfiniBand

Job description

NVIDIA is changing the world of AI Networking with groundbreaking technology. We are excited to be adding an AI Networking Software Developer to our AI Networking SW development and codesign team. We are working with the latest NVIDIA hardware and technologies. We do full stack benchmarking for Data Center scale systems for AI training and inference and lower level benchmarks. We strive for automation and develop many tools in-house yet adopt community accepted practices and frameworks. Our goal is to ensure that large-scale systems deliver expected performance in practice, not just on paper, by uncovering bottlenecks and driving continuous improvements.

What You'll Be Doing

• Develop AI networking communication frameworks and applications running in production on the world’s largest supercomputers and data centers.
• Develop production tools and benchmarks used by multiple teams inside and outside NVIDIA.
• Enable new AI models within our benchmarking infrastructure and deliver insights through end-to-end analysis of large-scale workloads across hardware and software stacks.
• Design and implement automation systems, including large-scale parameter search to identify optimal configurations across complex systems.
• Collaborate closely with networking and hardware teams to co-design new features and software interfaces in a fast-paced, evolving environment.

What We Need To See

• B.Sc., M.Sc degree in Computer Science / Software engineering or equivalent experience.
• 5+ years of experience.
• Professional Python development experience. We seek individuals who build maintainable, long-lived tools that do not impose a heavy burden on the team in terms of maintenance.
• Solid Linux expertise and passion for working extensively in command-line environments.
• Ability to work across a broad and evolving stack, with a strong drive to learn—from hardware and networking up to large-scale AI systems running across entire clusters.

Ways To Stand Out From The Crowd

• Knowledge and/or experience with modern AI ecosystem: PyTorch, LLMs, inference and training.
• Familiarity with cluster orchestration systems such as Slurm or Kubernetes.
• Knowledge in MPI and HPC, InfiniBand, Ethernet and Networking.
• Experience in performance optimizations.

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 152,000 USD - 241,500 USD for Level 3, and 184,000 USD - 287,500 USD for Level 4.

You will also be eligible for equity and benefits.

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.