Senior AI Networking & Performance Engineer

NVIDIA; Colorado; On-site; USD 272,000 - 431,250; Full time

Job summary

NVIDIA is seeking a Senior Software Engineer in AI Networking Codesign and Benchmarking R&D based in Colorado. The role involves profiling and optimizing AI workloads on large-scale GPU clusters, specifically tailored for distributed deep-learning training and inference.

This position requires expertise in high-performance networking and programming, with a strong emphasis on collaboration across hardware and software teams. A competitive salary ranging from $272,000 to $431,250, plus equity and benefits, is offered.

Qualifications

• 15+ years of experience with high-performance networking.
• Proficiency in programming languages: Python, Bash, C++.
• Experience with NVIDIA GPUs and deep-learning frameworks.

Responsibilities

• Profile, analyze, and optimize AI workloads on large-scale GPU and CPU clusters.
• Benchmark, profile, and analyze performance to find bottlenecks.
• Develop PyTorch trace-based profiling tools for benchmarking and debugging.

Skills

High-performance networking (RDMA, MPI, NCCL, SHARP)

Deep-learning frameworks (TensorFlow, PyTorch)

Programming languages (Python, Bash, C++)

Performance evaluation techniques

Communication skills

Education

B.Sc. in Computer Science or Software Engineering or equivalent

Tools

NVIDIA CUDA library

Container-based development environment

Job description

NVIDIA is seeking a Senior Software Engineer in AI Networking Codesign and Benchmarking R&D based in Colorado. The role involves profiling and optimizing AI workloads on large-scale GPU clusters, specifically tailored for distributed deep-learning training and inference.

This position requires expertise in high-performance networking and programming, with a strong emphasis on collaboration across hardware and software teams. A competitive salary ranging from $272,000 to $431,250, plus equity and benefits, is offered.