Senior High-Performance AI Training Engineer
Core
Optimizing AI training workloads across the full hardware/software stack to push performance to physical limits for NVIDIA's compute systems.
Role type
Senior High-Performance AI Training Engineer
Builds
Production-quality software for NVIDIA's deep learning platform stack, MLPerf Training benchmarks, and proprietary processor/system simulators.
Domain
AI/ML training, GPU architecture, high-performance computing
Deliverable
production ML models | infrastructure
Required skills
Deep learning and neural networks (training focus), computer architecture, GPU fundamentals, application performance tuning, processor/system-level performance modeling, C++, Python, CUDA
Preferred skills
None stated
Technologies
CUDA, C++, Python, MLPerf, proprietary simulators
Responsibilities
Profile and optimize AI training workloads on new platforms, solve performance issues across key training tasks, implement production software across driver to framework layers, build MLPerf submissions, develop tools to automate analysis and optimization workflows
Seniority
Senior, hands-on IC