NVIDIA · 3 hours ago
Senior AI Engineer, World Foundation Models
US, CA, Santa Clara
Full-time
Onsite
Senior Level, Lead/Staff
$184K/yr - $288K/yr
8+ years expNVIDIA is a leader in computer graphics and AI technology, seeking a Senior AI Engineer to advance world foundation models for high-fidelity video generation. The role involves research and development of model architectures and algorithms to enhance human-centric quality in video generation, ensuring real-time performance and user-visible quality through collaboration with various engineering teams.
AI InfrastructureArtificial Intelligence (AI)Consumer ElectronicsFoundational AIGPUHardwareSoftwareVirtual Reality
Responsibilities
Research, implement, and validate model architecture and algorithm changes that improve video generation fidelity, with emphasis on human-centric quality (identity preservation, anatomy, motion coherence, and interaction realism)
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
Qualification
Required
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 equivalent experience)
4+ 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 (e.g., batching, parallelism strategies, low-precision/quantization awareness, attention/KV-cache efficiency)
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
Preferred
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
Exposure to closed-loop training setups (e.g., reinforcement learning or preference-based optimization) for improving controllability, stability, and interaction quality in generated sequences
Benefits
Equity
Benefits
Company
NVIDIA
NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.
10001+ employees
H1B Sponsorship
NVIDIA has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1877)
2024 (1355)
2023 (976)
2022 (835)
2021 (601)
2020 (529)
Funding
Current Stage
Public CompanyTotal Funding
$4.09BKey Investors
ARPA-EARK Investment ManagementSoftBank Vision Fund
2023-05-09Grant· $5M
2022-08-09Post Ipo Equity· $65M
2021-02-18Post Ipo Equity
Leadership Team
Jensen Huang
Founder and CEO
Michael Kagan
Chief Technology Officer
Recent News
Company data provided by crunchbase