Senior Robotics Foundation Model Architect

Nvidia Corporation in; Santa Clara (CA); On-site; USD 152,000 - 288,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is seeking a hands-on Senior Solutions Architect for Robotics Foundation Model Training in Santa Clara. You will architect end-to-end training workflows for robotics foundation models, scale workloads across multi-GPU systems, and collaborate across researchers, engineers, and executives to shape Physical AI platforms.

The role emphasizes building reference architectures, optimizing data pipelines, and driving model efficiency with tools like PyTorch, NeMo, and JAX.

Qualifications

• MS, PhD, or equivalent experience in Computer Science, Artificial Intelligence, Electrical or Computer Engineering, Robotics, or a related field.

Responsibilities

• Architect and optimize end-to-end training workflows for robotics foundation models.
• Build proof-of-concepts, reference architectures, and agentic workflows to accelerate experimentation and model improvement.
• Scale pre-training, fine-tuning, and RL workloads across multi-GPU/multi-node systems.
• Identify and eliminate data pipeline bottlenecks for multimodal datasets (video, sensor data, trajectories).
• Collaborate with NVIDIA product and engineering teams to guide future Physical AI platforms.

Skills

Deep learning

Distributed computing

Robotics experience

Team collaboration

Communication skills

Education

MS/PhD or equivalent in CS/AI/EE/CE/Robotics

Tools

PyTorch

NVIDIA NeMo

JAX

Hugging Face Transformers

Job description

NVIDIA is seeking a hands-on Senior Solutions Architect for Robotics Foundation Model Training in Santa Clara. You will architect end-to-end training workflows for robotics foundation models, scale workloads across multi-GPU systems, and collaborate across researchers, engineers, and executives to shape Physical AI platforms.

The role emphasizes building reference architectures, optimizing data pipelines, and driving model efficiency with tools like PyTorch, NeMo, and JAX.