Senior AI Engineer, World Foundation Models

Details

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. Today the company is investing broadly in AI to define the next era of computing. The team is building next-generation AI systems that can perceive, reason about, and generate dynamic worlds. They advance world foundation models to enable high-fidelity, temporally stable video and world generation for Physical AI, simulation, and interactive experiences. This role operates at the applied-research boundary: developing and validating model improvements, then hardening them into production-grade checkpoints and recipes that teams can reliably build on. The technical focus is on human appearance, motion and action understanding. Work is delivered in close partnership with data, platform, and product engineering.

Responsibilities

• Research, implement, and validate 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.
• 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.
• 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.

Ways to stand out

• 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 (code generation, debugging, test creation, experiment automation, benchmark development, documentation, and large-codebase navigation).

Compensation & Benefits

• Base salary ranges (location- and level-dependent):
  • Level 4: 184,000 USD - 287,500 USD
  • Level 5: 224,000 USD - 356,500 USD
• Eligible for equity and benefits. See: https://www.nvidia.com/en-us/benefits/

Other details

• Applications accepted at least until May 9, 2026.
• NVIDIA uses AI tools in its recruiting processes.
• NVIDIA is an equal opportunity employer and values diversity in its workforce.