NVIDIA Corporation

Senior AI Training Performance Architect

NVIDIA Corporation$184K — $356K *
Enterprise Technology
5 - 7 years of experience
Job Overview by Ladders

Qualifications

  • PhD in Computer Science, Electrical Engineering, or Computer Engineering 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, notably GPU fundamentals.
  • Experience in analyzing and tuning application performance is essential.
  • Familiarity with processor and system-level performance modeling.
  • Proficiency in C++, Python, and CUDA is a must.

Responsibilities

  • Analyze, profile, and optimize AI training workloads on advanced hardware and software platforms.
  • Identify and solve performance bottlenecks of AI training on GPUs across key workloads.
  • Implement production-quality software across NVIDIA's deep learning platform stack.
  • Contribute to NVIDIA's submissions for MLPerf Training benchmarks.
  • Implement major deep learning training workloads in NVIDIA's proprietary simulators.
  • Develop tools to automate workload analysis and optimization processes.

Benefits

  • Eligible for equity and additional benefits.
  • Work with state-of-the-art hardware and software platforms.
  • Opportunity to contribute to cutting-edge AI training performance projects.
Full Job Description
Senior AI Training Performance Architect

What you will be doing:
  • Understand, analyze, profile, and optimize AI training workloads on state-of-the-art hardware and software platforms.
  • Identifying performance bottlenecks of AI training on GPUs, prioritizing and then solving problems across the key AI training workloads.
  • Implement production-quality software across multiple layers of NVIDIA's deep learning platform stack, from drivers to DL frameworks.
  • Build and support NVIDIA submissions for MLPerf Training benchmarks.
  • Implement key DL 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.


What we want to see:
  • 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 architecture 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.


Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5.

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until July 28, 2026.

This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.

About NVIDIA Corporation

Nvidia, a global leader in graphics, gaming, and AI technology, offers Nvidia careers and internship opportunities for those passionate about driving innovation in the tech industry. you'll find a company committed to growth, teamwork, and leadership in computer science and machine learning domains.

About Nvidia

A Pioneer in Technology and Innovation

Nvidia has cemented its reputation as a powerhouse in developing advanced graphics processing units (GPUs) and has significantly contributed to the gaming industry's evolution. Moreover, its foray into AI and machine learning has opened new frontiers in technology, making Nvidia a beacon of innovation and a desirable workplace for ambitious tech professionals.

Job Opportunities

Diverse Positions in a Dynamic Field

Nvidia is continuously on the lookout for talented individuals across various domains, including hardware and software engineering, product design, marketing, and sales. Employment opportunities at Nvidia are vast, catering to a wide range of expertise and career aspirations.

Employment in Hardware and Graphics

For those fascinated by the intricacies of hardware and graphics technology, Nvidia offers positions that sit at the forefront of gaming and computing advancements.

Growth in Machine Learning and AI

Nvidia's leadership in AI and machine learning has created numerous vacancies for specialists eager to contribute to groundbreaking projects.

Recruitment in Computer Science

With the constant demand for innovation, Nvidia's recruitment efforts focus on computer science experts capable of pushing the boundaries of what's possible.

Internship Program

Opening Doors to Future Innovators

Nvidia's internship program is designed to nurture the next generation of technology leaders, offering hands-on experience in a culture that celebrates creativity and teamwork.

Benefits and Culture

Interns at Nvidia enjoy a plethora of benefits, from competitive stipends to mentorship opportunities, all within an environment that values growth and learning.

Opportunities for Students

Whether you're an undergraduate, a master's student, or a Ph.D. candidate, Nvidia's internships provide a real-world glimpse into the tech industry, offering valuable experience in various technology fields.

Pathways to Full-Time Employment

Many interns have transitioned into full-time positions, marking the start of successful careers at Nvidia. The internship program is more than a stepping stone into the company; it’s an investment in the professional development of interns. The goal is to ensure that interns are well-equipped for future challenges.

Nvidia Careers: More Than Just a Job

Nvidia offers more than just a job to its employees; it provides a front-row seat on the journey into the future of technology. Nvidia stands as a pillar of innovation with its vast opportunities in hardware, graphics, gaming, machine learning, and computer science. Nvidia careers serve as a launching pad for talented workers who aim to redefine the technological landscape. Whether through full-time positions or internships, joining Nvidia means contributing to a legacy of breakthroughs and becoming part of a global community dedicated to pushing the boundaries of what's possible.
Learn more about NVIDIA Corporation
Size
22,473 employees
Market Cap
$350.4 billion
Industry
Net Income
$4.3 billion
Founded
1993
5 Year Trend
+31.3%
Revenue
$16.6 billion
NASDAQ

Similar Jobs

More Jobs at NVIDIA Corporation

More Enterprise Technology Jobs

Find similar Senior AI Training Performance Architect jobs: