Senior AI Engineer: Real-Time Gaming & World Models

NVIDIA; United States; Remote; USD 224,000 - 431,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA Lightspeed Studios seeks a Senior Software Engineer to push the boundaries of AI and graphics for upcoming game projects. You will help bring world models and video diffusion techniques into real-time gameplay, from prototype to release, while mentoring engineers and shaping technical decisions.

The role prioritizes hands-on model training, performance optimization, and seamless engine integration across Unreal, Unity, and custom engines, with collaboration across research, rendering, and

Qualifications

• BS, MS, or PhD in Computer Science, Electrical Engineering, or a related field, or equivalent experience.
• 12+ years of software engineering experience, including substantial hands-on work shipping AI, deep learning, or machine learning systems.
• Expert-level C++ and Python.
• Hands-on experience training and fine-tuning deep learning models with PyTorch or a similar framework, including distributed training.
• Experience in video generation, world models, diffusion or flow-matching models, transformers, or neural rendering.
• Experience optimizing GPU inference with CUDA, TensorRT, TensorRT-LLM, or Triton.
• A track record of turning complex prototypes into shipped products, and clear communication across research, engineering, and art.

Responsibilities

• Build and ship software for upcoming, not-yet-announced projects that bring world models and video diffusion models to real-time gaming, from prototype to release.
• Train, fine-tune, and evaluate models, including data curation and adapting models to game-specific content and controls.
• Optimize models and inference for latency, token throughput, and quality, using techniques such as distillation, quantization, and reduced-step sampling on RTX devices and in the cloud.
• Profile and remove bottlenecks across the stack, from model architecture to GPU kernels.
• Integrate models into Unreal Engine, Unity, and custom engines, so they run efficiently alongside rendering.
• Lead technical decisions, mentor other engineers, and collaborate with NVIDIA Research, rendering, and art teams.

Skills

C++

Python

CUDA

TensorRT

TensorRT-LLM

Triton

Education

BS/MS/PhD in CS/EE

Tools

PyTorch

Job description

NVIDIA Lightspeed Studios seeks a Senior Software Engineer to push the boundaries of AI and graphics for upcoming game projects. You will help bring world models and video diffusion techniques into real-time gameplay, from prototype to release, while mentoring engineers and shaping technical decisions.

The role prioritizes hands-on model training, performance optimization, and seamless engine integration across Unreal, Unity, and custom engines, with collaboration across research, rendering, and