Senior 3D Spatial AI Engineer — Data & Training Lead

NVIDIA; Santa Clara (CA); On-site; USD 224,000 - 357,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA, a leader in AI and graphics, seeks a Senior Deep Learning Engineer to own the data engine and end-to-end training verification for 3D spatial reasoning in Cosmos Lab. You will curate large-scale real-world image/video corpora, build 3D-grounded supervision, and ensure reproducible pre-training and fine-tuning pipelines across multi-GPU clusters.

You will collaborate with Cosmos Lab researchers to translate 3D hypotheses into measurable capabilities in shipped models, drive data quality,

Qualifications

• 12+ years of proven experience building deep learning systems in Python with PyTorch or JAX on Linux.
• The candidate has hands-on experience with vision-language models and data pipelines for large-scale multimodal data.
• Experience running and validating large model training on multi-GPU, multi-node clusters with distributed training strategies.

Responsibilities

• Own the 3D data engine and end-to-end training verification for 3D spatial reasoning.
• Build and operate 3D and spatial evaluation suites and benchmarks with traceability.

Skills

3D computer vision

PyTorch

JAX

Distributed training

Multi-node Kubernetes/Rain

Communication

Education

MS or PhD in CS/Engineering/Robotics

Tools

Python

Linux

PyTorch

JAX

Job description

NVIDIA, a leader in AI and graphics, seeks a Senior Deep Learning Engineer to own the data engine and end-to-end training verification for 3D spatial reasoning in Cosmos Lab. You will curate large-scale real-world image/video corpora, build 3D-grounded supervision, and ensure reproducible pre-training and fine-tuning pipelines across multi-GPU clusters.

You will collaborate with Cosmos Lab researchers to translate 3D hypotheses into measurable capabilities in shipped models, drive data quality,