Lead High-Performance LLM Training Architect (Equity)

NVIDIA; Santa Clara (CA); On-site; USD 272,000 - 431,250; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is looking for a Principal Engineer in Santa Clara to enhance performance across AI training and post-training workloads. The role requires significant expertise in GPU architecture and deep learning frameworks, focusing on optimizing large-scale workloads and mentoring engineers.

The ideal candidate will possess a PhD or MS with over 12 years of experience in relevant fields. With a base salary range of $272,000 to $431,250, you will play a pivotal role in shaping the future of AI.

Qualifications

• 12+ years of relevant work or research experience in AI training.
• Hands-on experience with LLM pre-training and reinforcement learning.
• Strong understanding of GPU and AI accelerator architecture.

Responsibilities

• Lead performance analysis and optimization of AI workloads.
• Identify and remove bottlenecks to improve performance.
• Mentor engineers in AI performance analysis best practices.

Skills

Deep learning frameworks

Performance engineering

GPU architecture

Distributed training techniques

Performance modeling tools

Education

MS or PhD in Computer Science or related field

Tools

CUDA libraries

Job description

NVIDIA is looking for a Principal Engineer in Santa Clara to enhance performance across AI training and post-training workloads. The role requires significant expertise in GPU architecture and deep learning frameworks, focusing on optimizing large-scale workloads and mentoring engineers.

The ideal candidate will possess a PhD or MS with over 12 years of experience in relevant fields. With a base salary range of $272,000 to $431,250, you will play a pivotal role in shaping the future of AI.