Taro Logo

Senior High-Performance AI Training Engineer

NVIDIA is the world leader in accelerated computing, pioneering AI and digital twins technology.
$184,000 - $356,500
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
8+ years of experience
AI

Description For Senior High-Performance AI Training Engineer

NVIDIA, a pioneer in accelerated computing for over 25 years, is seeking a Senior High-Performance AI Training Engineer to join their innovative team. This role sits at the intersection of AI and high-performance computing, focusing on optimizing AI training workloads for maximum efficiency. The position offers a unique opportunity to work with cutting-edge technology and directly impact the hardware and software roadmap of a leading AI technology company.

The ideal candidate will be deeply involved in performance analysis and optimization, working across all layers of the hardware/software stack - from GPU architecture to application code. They will be responsible for analyzing and optimizing AI training workloads, implementing production-quality software, and contributing to NVIDIA's MLPerf Training benchmarks. This role requires a strong background in deep learning, computer architecture, and performance optimization.

NVIDIA offers a comprehensive compensation package, including a competitive base salary range of $184,000 - $356,500, equity, and full benefits. The company is known for its innovative culture and commitment to pushing the boundaries of what's possible in AI and accelerated computing. As an NVIDIAN, you'll be part of a diverse, supportive environment where everyone is inspired to do their best work.

The position is based in Santa Clara, CA, and requires either a PhD with 5+ years of experience or an MS with 8+ years of experience. This is an excellent opportunity for someone passionate about high-performance computing and AI to make a lasting impact on the technology that's shaping our future.

Last updated 9 hours ago

Responsibilities For Senior High-Performance AI Training Engineer

  • Understand, analyze, profile, and optimize AI training workloads on new hardware and software platforms
  • Prioritize and solve performance issues across all new neural networks
  • Implement production-quality software across multiple layers of NVIDIA's deep learning platform stack
  • Build and support NVIDIA submissions for MLPerf Training benchmarks
  • Implement key DL training workloads in NVIDIA's proprietary processor and system simulators
  • Develop tools to automate workload analysis, optimization, and other critical workflows

Requirements For Senior High-Performance AI Training Engineer

Python
Linux
  • PhD in CS, EE or CSEE with 5+ years experience or MS with 8+ years experience
  • Strong background in deep learning and neural networks, particularly in training
  • Solid understanding of computer architecture and 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

Benefits For Senior High-Performance AI Training Engineer

Medical Insurance
Dental Insurance
Vision Insurance
Equity
  • Competitive salaries
  • Equity
  • Comprehensive benefits package

Interested in this job?

Jobs Related To NVIDIA Senior High-Performance AI Training Engineer