Senior AI Systems Tools Engineer - GPU Clusters

NVIDIA Gruppe; Santa Clara (CA); Hybrid; USD 184,000 - 357,000; Full time

Job summary

NVIDIA is seeking engineers to develop tools for AI researchers and software/hardware teams running AI workloads on GPU clusters. You will build profiling, analysis, and debugging tools at scale.

The role emphasizes strong C++/Python skills, experience with PyTorch/TensorFlow, and collaboration with hardware architects to deliver impactful features and improvements. Hybrid work is supported in our global setup.

Qualifications

• BS+ in Computer Science or related field (or equivalent experience) and 6+ years of software development.
• Strong skills in design, coding (C++ and Python), analytical thinking and debugging.
• Knowledge of PyTorch and TensorFlow; distributed training and inference.

Responsibilities

• Build internal profiling and analysis tools for AI workloads at large scale.
• Build debugging tools for memory, networking, and other common problems.
• Create benchmarking and simulation technologies for AI systems or GPU clusters.
• Partner with HW architects to propose new features or improve existing ones with real-world use cases.

Skills

C++

Python

Debugging

Distributed training

Problem solving

Education

BS in Computer Science or related

Tools

PyTorch

TensorFlow

Slurm

Kubernetes

CUDA

NCCL

Linux

Job description

NVIDIA is seeking engineers to develop tools for AI researchers and software/hardware teams running AI workloads on GPU clusters. You will build profiling, analysis, and debugging tools at scale.

The role emphasizes strong C++/Python skills, experience with PyTorch/TensorFlow, and collaboration with hardware architects to deliver impactful features and improvements. Hybrid work is supported in our global setup.