Senior High-Performance AI Training Engineer
NVIDIA · Santa Clara, CA · 1 wk ago
EngineeringFull-time
About the role
NVIDIA is seeking senior engineers with a passion for performance analysis and optimization to work on AI training workloads. This role involves working across various layers of the hardware and software stack to push the limits of performance.
Responsibilities
- Understand, analyze, profile, and optimize AI training workloads on new hardware and software platforms.
- Prioritize and solve performance issues across key AI model training tasks.
- Implement production-quality software across multiple layers of NVIDIA's deep learning platform stack, including drivers and DL frameworks.
- Build and support NVIDIA submissions for MLPerf Training benchmarks.
- Develop tools to automate workload analysis, optimization, and other critical workflows.
Requirements
- PhD in CS, EE or CSEE 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 GPU 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.
Qualifications
- Widely considered to be one of the technology world’s most desirable employers.
- Competitive compensation, including base salary ranging from $184,000 to $287,500 for Level 4, and $224,000 to $356,500 for Level 5.
- Equity and comprehensive benefits package.
Benefits
- NVIDIA offers a wide range of benefits, including health insurance, retirement plans, and paid time off.
Pay
Base salary range: $184,000 - $287,500 for Level 4, and $224,000 - $356,500 for Level 5.
Schedule
Full-time position.
Skills
- Experience with deep learning and neural networks.
- Knowledge of computer architecture and GPU fundamentals.
- Proficiency in C++, Python, and CUDA.
- Experience with performance modeling and optimization.
Benefits
- Comprehensive benefits package including health insurance, retirement plans, and paid time off.
- Equity options.
Application Instructions
Applications for this job will be accepted at least until February 16, 2026.