Drimva

NVIDIA

Senior AI Training Performance Architect

$184K–$356K

US, CA, Santa Clara · Senior · Full-time · Posted Jul 24, 2026

Experience
PhD in CS, EE or CSEE (or equivalent experience) with 5+ years of relevant experience; or MS with 8+ years of experience.
Education
PhD in CS, EE or CSEE (or equivalent experience); or MS.

About the job

Analyzes and optimizes AI training performance across NVIDIA’s hardware and software stack, including workloads, benchmarks, tools, and architecture studies.

What you'll do

  • Profile and optimize AI training workloads; identify and resolve GPU performance bottlenecks.
  • Implement production-quality software across NVIDIA’s deep learning stack and support MLPerf Training submissions.
  • Implement training workloads in processor and system simulators and develop workflow automation tools.

What they ask for

  • Strong background in deep learning and neural network training.
  • Understanding of computer architecture and familiarity with GPU architecture fundamentals.
  • Proven experience analyzing and tuning application performance, and with processor and system-level performance modeling.
  • Proficiency in C++, Python, and CUDA.

Pay

184,000 USD - 287,500 USD (base salary, Level 4); 224,000 USD - 356,500 USD (base salary, Level 5)

Tools

C++, Python, CUDA, GPUs, MLPerf Training

AI focus

Optimizes AI training workloads and develops benchmark, simulation, and analysis capabilities for deep learning systems.

Model-written summary · the original posting is authoritative

First seen Oct 8, 2026 · Last verified Nov 5, 2026 (UTC)