Senior AI Training Solutions Architect (Foundation Models)

NVIDIA Corporation; Seattle (WA); On-site; USD 152,000 - 242,000; Full time

Job summary

NVIDIA Corporation is seeking an experienced Senior Solutions Architect to help customers build, scale, and optimize AI training solutions using NVIDIA hardware and software on cloud platforms. The role emphasizes foundation model pre-training and post-training, with reinforcement learning focus, engaging with researchers, engineers, and developers across our strategic customers.

You will design end-to-end training workflows, optimize multi-GPU training, and diagnose bottlenecks in compute,

Qualifications

• BS, MS, or PhD in Computer Science, Electrical or Computer Engineering, Statistics, Physics, or a related technical field, or equivalent experience.
• 5+ years of relevant professional experience in solutions architecture, software engineering, ML engineering, research engineering, or related technical roles, including 3+ years of hands-on experience with large-scale AI systems or distributed training.
• Strong programming and debugging skills in Python, with hands-on experience developing and optimizing deep learning workloads using PyTorch.

Responsibilities

• Work with Cloud Service Providers to design, build, and demonstrate scalable AI training solutions using NVIDIA hardware and software.
• Lead technical engagements across foundation model pre-training and post-training, helping customers move from architecture and proofs of concept to production adoption.
• Develop and optimize end-to-end training workflows spanning data preparation, distributed training, checkpointing, and evaluation, including reinforcement learning pipelines.
• Build customer-specific proofs of concept that address critical technical and business needs, demonstrating improvements in training performance, reliability, and efficiency.
• Help customers diagnose and resolve bottlenecks across GPU compute, memory, communication, networking, storage, and workload orchestration.

Skills

Python

PyTorch

Distributed training

Multi-GPU training

Cloud platforms

Training workflows

Education

BS/MS/PhD in CS or related field

Tools

NVIDIA NeMo

Megatron Core

Megatron Bridge

PyTorch FSDP

DeepSpeed

Job description

NVIDIA Corporation is seeking an experienced Senior Solutions Architect to help customers build, scale, and optimize AI training solutions using NVIDIA hardware and software on cloud platforms. The role emphasizes foundation model pre-training and post-training, with reinforcement learning focus, engaging with researchers, engineers, and developers across our strategic customers.

You will design end-to-end training workflows, optimize multi-GPU training, and diagnose bottlenecks in compute,