Senior AI Researcher - World Foundation Models

About the role

NVIDIA is seeking a researcher to develop and validate model architecture and algorithm changes that improve video generation fidelity, with a focus on human-centric quality. This role involves exploring and prototyping improvements across various modalities and techniques, and translating research results into robust implementations.

Responsibilities

• Research and implement model architecture and algorithm changes that improve video generation fidelity, with emphasis on human-centric quality.
• Explore and prototype improvements across spatial multimodal modeling, modality alignment, flow-based or diffusion-based video generation, and neural rendering-inspired representations to improve controllability and long-horizon consistency.
• Improve training and inference efficiency through architectural and post-training techniques (compute/memory optimizations, distillation, pruning, and compression).
• Define model training objectives that improve sim-to-real and real-to-sim generalization, especially for human motion, contact, and interaction dynamics across real-world and synthetic/simulation data.
• Develop detailed, domain-specific benchmarks for evaluating world foundation models, especially generation and understanding world models that reason about video, simulation, and physical environments.
• Translate research results into robust implementations like training code, production-grade checkpoints, model integrations, and demos that clearly showcase capability gains across teams.

Requirements

• PhD in Computer Science, Graphics, Computer Engineering, or a closely related field (or equivalent experience).
• 8+ years of applied research and/or industry experience in vision, graphics, or adjacent ML domains or similar area.
• 3+ years of direct experience designing, training, and evaluating generative models for image/video/audio, with strong fundamentals in modern deep learning.
• Hands-on experience improving generative models with a focus on perceptual quality and temporal stability, especially for generating humans.
• Advanced proficiency in Python, PyTorch, C++, and CUDA with strong research-engineering practices (reproducibility, testing, profiling, experiment tracking).
• Experience training and debugging large models in multi-GPU and/or multi-node environments and distributed training workflows.
• Practical knowledge of inference/runtime bottlenecks and optimization techniques.
• A strong "eye for quality" and interest in diagnosing visual artifacts (sharpness, texture detail, temporal stability, etc.) using perceptual metrics, human preference signals, or learned evaluators.

Qualifications

• Proven track record in related research, including publications in top conferences (e.g., NeurIPS, CVPR, ICLR), with clear evidence of impact on model quality or robustness.
• Experience using agentic workflows, and AI coding companions, to accelerate research and production development, including code generation, debugging, test creation, experiment automation, benchmark development, documentation, and large-codebase navigation.

Skills

• Strong background in computer science, computer vision, machine learning, and related fields.
• Expertise in deep learning frameworks such as PyTorch and TensorFlow.
• Experience with distributed training and optimization techniques.
• Ability to design and evaluate complex models and algorithms.
• Excellent communication and collaboration skills.

Benefits

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 to you and your family at www.nvidiabenefits.com/. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is $184,000 - $287,500 for Level 4, and $224,000 - $356,500 for Level 5. You will also be eligible for equity and benefits.

Pay

$184,000 - $287,500 for Level 4
$224,000 - $356,500 for Level 5

Schedule

Full-time