The Senior AI Training Performance Architect at NVIDIA is focused on performance analysis and optimization of AI training workloads. This role involves working across the hardware/software stack to enhance the performance of NVIDIA's deep learning platform. The position offers the chance to influence the hardware and software roadmap in a leading technology company. Candidates will be part of a team that is at the forefront of the AI revolution, contributing to the design and construction of powerful compute systems.

• PhD in Computer Science, Electrical Engineering, or Computer Engineering with 5+ years of relevant experience, or a Master's degree 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.

• Understand, analyze, profile, and optimize AI training workloads on state-of-the-art hardware and software platforms.
• Identify performance bottlenecks of AI training on GPUs and prioritize and solve problems across key AI training workloads.
• 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.

• Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.