# Senior AI Training Performance Architect

**Company**: NVIDIA
**Location**: Santa Clara, CA
**Experience**: senior
**Job type**: full-time
**Category**: Engineering
**Industry**: Technology

**Apply**: https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/US-CA-Santa-Clara/Senior-AI-Training-Performance-Architect_JR2021938?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply
**Canonical**: https://yubhub.co/jobs/job_2b9eda83-23a

## Description

NVIDIA is seeking a senior engineer who is obsessed with performance analysis and optimization to help us squeeze every last clock cycle out of AI training, the workload driving the design and construction of the largest and most powerful compute systems in the world.

You will be working across all layers of the hardware/software stack - from GPU architecture to the application code - to achieve peak performance. This role offers the opportunity to directly impact the hardware and software roadmap in a fast-growing technology company that leads the AI revolution.

### Responsibilities

- 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, 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 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.

### 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 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.

### Benefits

You will also be eligible for equity and benefits.

## Skills

### Required
- C++
- Python
- CUDA
- deep learning
- neural networks
- computer architecture
- GPU architecture
- performance analysis
- optimization

---

Source: [Apply at nvidia.wd5.myworkdayjobs.com](https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/US-CA-Santa-Clara/Senior-AI-Training-Performance-Architect_JR2021938?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply)