Senior AI Performance Tools Engineer (GPU Profiling)

NVIDIA; United States; On-site; USD 184,000 - 287,500; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is looking for a skilled software engineer to develop internal profiling and debugging tools for AI workloads in a GPU cluster. The ideal candidate will have over 6 years of experience, with strong skills in C++ and Python, and a solid understanding of deep learning frameworks like PyTorch and TensorFlow.

The position is based in the United States, with a competitive salary range of $184,000 to $287,500, depending on experience and level. NVIDIA is committed to diversity and equal opportunity in its hiring practices.

Qualifications

• 6+ years of software development experience.
• Strong design and coding skills in C++ and Python.
• Good understanding of AI systems and GPU architecture.

Responsibilities

• Build internal profiling and analysis tools for AI workloads.
• Create debugging tools for memory and networking issues.
• Develop benchmarking technologies for AI systems.

Skills

C++

Python

Deep Learning frameworks (PyTorch, TensorFlow)

GPU cluster job scheduling (Slurm or Kubernetes)

NVIDIA GPUs

CUDA Programming

Analytical skills

Debugging

Education

BS in Computer Science or related field

Job description

NVIDIA is looking for a skilled software engineer to develop internal profiling and debugging tools for AI workloads in a GPU cluster. The ideal candidate will have over 6 years of experience, with strong skills in C++ and Python, and a solid understanding of deep learning frameworks like PyTorch and TensorFlow.

The position is based in the United States, with a competitive salary range of $184,000 to $287,500, depending on experience and level. NVIDIA is committed to diversity and equal opportunity in its hiring practices.