Senior Software Developer, AI Networking

Stack

Responsibilities

• ▹Develop AI networking communication frameworks and applications running in production on large-scale supercomputers and data centers
• ▹Develop production tools and benchmarks used by multiple teams inside and outside NVIDIA
• ▹Enable new AI models within the benchmarking infrastructure
• ▹Deliver insights through end-to-end analysis of large-scale workloads
• ▹Design and implement automation systems, including large-scale parameter search
• ▹Collaborate closely with networking and hardware teams to co-design new features

Requirements

• ▹BSc or MSc in Computer Science / Software Engineering
• ▹5+ years of experience or equivalent
• ▹Professional Python development experience
• ▹Ability to build maintainable, long-lived tools
• ▹Solid Linux expertise and command-line proficiency
• ▹Ability to work across a broad, evolving stack

Nice to have

• ▹Knowledge of the modern AI ecosystem: PyTorch, LLMs, inference and training
• ▹Familiarity with cluster orchestration systems such as Slurm or Kubernetes
• ▹Knowledge of MPI, HPC, InfiniBand, Ethernet and networking
• ▹Experience with performance optimization

Soft skills

About the company

NVIDIA pioneered visual computing with the invention of the GPU and is now at the forefront of the AI computing era.