Senior 3D Spatial AI Engineer — Data & Training Lead

NVIDIA; Santa Clara (CA); In loco; USD 224.000 - 357.000; Tempo pieno

Vantaggi offerti da questo lavoro

Equity

Benefits

Descrizione del lavoro

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,

Competenze

• 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.

Mansioni

• 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.

Conoscenze

3D computer vision

PyTorch

JAX

Distributed training

Multi-node Kubernetes/Rain

Communication

Formazione

MS or PhD in CS/Engineering/Robotics

Strumenti

Python

Linux

PyTorch

JAX

Descrizione del lavoro

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,