Senior AI Networking & Performance Engineer

NVIDIA; Town of Texas (WI); On-site; USD 272,000 - 431,250; Full time

Job summary

NVIDIA is seeking a Senior Software Engineer to join the AI Networking Codesign and Benchmarking R&D group. In this role, you will analyze and optimize AI workloads on large-scale GPU and CPU clusters, focusing on collective communication and networking.

The ideal candidate will have a B.Sc. in Computer Science or Software Engineering and 15+ years of experience with high-performance networking. This position offers a competitive salary and the opportunity to work with cutting-edge AI technologies.

Qualifications

• 15+ years of experience with high-performance networking techniques.
• Demonstrated ability in performance evaluation techniques and approaches.
• Experience with a container-based development environment.

Responsibilities

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

Skills

High-performance networking

NVIDIA GPUs

Programming in Python

CUDA libraries

Deep learning frameworks (TensorFlow, PyTorch)

Analytical and problem-solving skills

Education

B.Sc. in Computer Science or Software Engineering

Job description

NVIDIA is seeking a Senior Software Engineer to join the AI Networking Codesign and Benchmarking R&D group. In this role, you will analyze and optimize AI workloads on large-scale GPU and CPU clusters, focusing on collective communication and networking.

The ideal candidate will have a B.Sc. in Computer Science or Software Engineering and 15+ years of experience with high-performance networking. This position offers a competitive salary and the opportunity to work with cutting-edge AI technologies.