Senior AI Networking Software Engineer — Benchmarking & HPC

NVIDIA; Town of Texas (WI); On-site; USD 152,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 develop AI networking frameworks and production tools for large-scale data center workloads, and contribute to end-to-end benchmarking across hardware and software stacks.

You will collaborate with networking and hardware teams to co-design features, automate large-scale experiments, and analyze performance to drive improvements on production systems.

Qualifications

• B.Sc. or M.Sc degree in Computer Science / Software engineering or equivalent experience.
• 5+ years of experience.
• Professional Python development experience with maintainable tooling.
• Solid Linux expertise and command-line proficiency.
• Ability to learn across hardware, networking, and large-scale AI systems.

Responsibilities

• 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.

Skills

Python development

Linux expertise

Broad stack learning

Education

B.Sc. or M.Sc in Computer Science / Software Engineering

Tools

Slurm

Kubernetes

MPI

InfiniBand

Job description

NVIDIA is seeking an AI Networking Software Developer to join our AI Networking SW development and codesign team. You will develop AI networking frameworks and production tools for large-scale data center workloads, and contribute to end-to-end benchmarking across hardware and software stacks.

You will collaborate with networking and hardware teams to co-design features, automate large-scale experiments, and analyze performance to drive improvements on production systems.