Senior AI Training Architect - Distributed Systems

NVIDIA Corporation; United States; Remote; USD 180,000 - 280,000; Full time

Job summary

NVIDIA Corporation is seeking a Senior Solutions Architect to support large-scale AI training and model alignment initiatives across global customers, including research institutions. You will bridge research and engineering, guiding customers on the full training stack and infrastructure needs.

You will lead strategic technical engagements, optimize training efficiency on multi-node GPU clusters, and work with Megatron-LM, NeMo and PyTorch.

Qualifications

• MS/PhD or equivalent experience in AI, ML or systems
• 7+ years in distributed AI training with multi-node GPUs
• Strong knowledge of training infrastructure and efficiency

Responsibilities

• Build and manage strategic technical relationships with leading AI model builders across EMEA
• Define software stack and infrastructure for large-scale training and post-training workflows
• Advise customers on distributed training strategies and resource optimization
• Coordinate with product and research teams and support developer communities

Skills

Distributed AI training

Megatron-LM / NeMo

HPC clusters

Excellent communication

Education

MS or PhD in Computer Science, Engineering or equivalent

Tools

Megatron-LM

NeMo

PyTorch

Job description

NVIDIA Corporation is seeking a Senior Solutions Architect to support large-scale AI training and model alignment initiatives across global customers, including research institutions. You will bridge research and engineering, guiding customers on the full training stack and infrastructure needs.

You will lead strategic technical engagements, optimize training efficiency on multi-node GPU clusters, and work with Megatron-LM, NeMo and PyTorch.