Solutions Architect – Large Scale AI Training

NVIDIA Corporation; United States; Remote; USD 150,000 - 210,000; Full time

Benefits offered by this job

Competitive salary

Comprehensive benefits package

Job summary

NVIDIA Corporation is seeking a senior Solutions Architect with deep expertise in large-scale distributed training and alignment of neural networks. You will support leading AI researchers and institutions in training and post-training workflows, including MoE models, and bridge research with engineering across NVIDIA’s full training stack.

As the primary technical expert, you will guide customers on software stacks and infrastructure, optimize GPU utilization and memory at scale, and

Qualifications

• MS or PhD in Computer Science, Engineering, or equivalent.
• 7+ years of distributed AI training experience on multi-node GPU clusters.
• Strong understanding of training infrastructure, efficiency and scalability.
• Excellent communication with researchers and infrastructure engineers.

Responsibilities

• Build and manage strategic technical relationships with leading AI model builders in EMEA.
• Collaborate with customers to define software stacks and infrastructure for large-scale training and post-training workflows, incl. RL.
• Serve as the go-to expert on distributed training strategies, guiding customers through large-scale training and post-training recipes using frameworks like Megatron-LM, NeMo, or PyTorch.
• Help customers optimize training/finetuning efficiency at scale, covering GPU utilization, memory management, and communication overlap.
• Collaborate with NVIDIA product and research teams to present customer needs.
• Animate the developer community by building or supporting hackathons, demos, and technical conferences.
• Publish or contribute to open-source work in distributed training.

Skills

Distributed training

Megatron-LM

NeMo

Reinforcement Learning

RLHF

High-Performance Computing

Communication skills

Education

MS or PhD in Computer Science/Engineering

Tools

PyTorch

NeMo framework

Megatron-LM framework

Job description

Описание:

We are looking for a senior Solutions Architect with strong knowledge of large-scale distributed training and Alignment of Neural Networks. The position involves supporting top EMEA AI Natives and research institutions in training/post training AI models, such as large Mixture-of-Experts (MoE) models. In this role, you will be the primary technical expert for AI organizations looking at some of the hardest infrastructure and algorithmic challenges in model training and alignment. You will bridge research and engineering, guiding customers on NVIDIA's full training stack (including Nemotron) while crafting NVIDIA product roadmap based on customers' feedback.

Задачи:

Build and manage strategic technical relationship with leading EMEA AI model builders developing large-scale foundation models.

Collaborate closely with customers to define the software stack and infrastructure required for large-scale training and post-training workflows, including Reinforcement Learning (RL).

Serve as the go-to expert on distributed training strategies, guiding customers through efficient large-scale training and post training recipes using Deep Learning Frameworks such as PyTorch, Megatron-LM or NeMo (RL/Gym).

Help customers optimize training/finetuning efficiency at scale, covering GPU utilization, communication overlap, memory management at scale.

Collaborate with NVIDIA's product and research teams to present customer needs.

Animate the developer community by building or supporting hackathons, demos, technical conferences.

What we need to see: MS or PhD in Computer Science, Engineering, or equivalent experience. Over 7 years of practical experience in distributed AI training, including direct involvement with HPC and/or AI environments with multi-node GPU clusters. Solid understanding of training infrastructure and how it affects efficiency and scalability. Strong proficiency with Megatron-LM, NeMo, or equivalent distributed training frameworks. Excellent communication skills with an ability to engage both research scientists and infrastructure engineers.

Требования:

Experience in fine-tuning with Reinforcement Learning (RLVR, RLHF) at scale.

Experience with LatentMoE, expert load balancing, and speculative decoding for MoE inference.

Prior experience in an AI Datacenter/HPC center, national lab, or frontier AI lab environment.

Published work or open-source contributions in distributed training.

Widely considered to be one of the technology world’s most desirable employers, NVIDIA offers highly competitive salaries and a comprehensive benefits package.

As you plan your future, see what we can offer www.nvidiabenefits.com/ NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer.

As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.