Senior High-Performance AI Training Engineer

at Nvidia
USD 184,000-356,500 per year
SENIOR
✅ On-site

Used Tools & Technologies

Not specified

Required Skills & Competences

Python @ 6 CUDA @ 6 GPU @ 3 Deep Learning @ 4 AI @ 4 Performance Analysis @ 4

Details

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. Today, NVIDIA is tapping into the unlimited potential of AI to define the next era of computing. This role is for senior engineers obsessed with performance analysis and optimization to help squeeze every last clock cycle out of AI training across hardware and software stacks.

Responsibilities

  • Understand, analyze, profile, and optimize AI training workloads on new hardware and software platforms, identifying fundamental performance limiters.
  • Prioritize and solve performance issues across key AI model training tasks, pushing end-to-end performance toward physical limits.
  • Implement production-quality software across multiple layers of NVIDIA's deep learning platform stack, from drivers to deep learning frameworks.
  • Build and support NVIDIA submissions for MLPerf Training benchmarks.
  • Implement key deep learning training workloads in NVIDIA's proprietary processor and system simulators to enable future architecture studies.
  • Develop tools to automate workload analysis, optimization, and other critical workflows.

Requirements

  • PhD in CS, EE or CSEE (or equivalent experience) with 5+ years of relevant experience; or MS with 8+ years of experience.
  • Strong background in deep learning and neural networks, particularly in training.
  • Solid understanding of computer architecture and familiarity with GPU fundamentals.
  • Proven background in analyzing and tuning application performance.
  • Proven experience with processor and system-level performance modeling.
  • Proficiency in programming with C++, Python, and CUDA.

Compensation and Benefits

Additional Information

  • Applications for this job will be accepted at least until February 16, 2026.
  • This posting is for an existing vacancy.
  • NVIDIA uses AI tools in its recruiting processes.
  • NVIDIA is an equal opportunity employer and does not discriminate on the basis of protected characteristics.