Senior AI Training Performance Engineer
Core
Analyze, profile, and optimize AI and deep learning training workloads on state-of-the-art hardware and software platforms to achieve peak performance.
Role type
Senior IC AI training performance engineer
Builds
Production-quality software for NVIDIA's deep learning platform stack (drivers to DL frameworks) and tools for workload analysis/optimization
Domain
AI Computing / Deep Learning / GPU Architecture
Deliverable
production ML models
Required skills
Deep learning and neural networks training, computer architecture, GPU architecture fundamentals, application performance tuning, processor and system-level performance modelling, C++, Python, CUDA
Preferred skills
None stated
Technologies
CUDA, Deep Learning Frameworks, GPU simulators
Responsibilities
Understand, analyze, profile, and optimize AI and deep learning training workloads; Implement production-quality software in multiple layers of NVIDIA's deep learning platform stack; Implement key DL training workloads in NVIDIA's proprietary processor and system simulators; Build tools to automate workload analysis, workload optimization, and other critical workflows
Seniority
Senior, hands-on IC
