AI Software Engineer, Lightspeed Studios

NVIDIA is a technology company whose Lightspeed Studios team combines AI and graphics technology to advance the future of gaming. The Senior Software Engineer will develop, train, optimize, and integrate AI models into real-time gaming experiences, while leading technical decisions and collaborating across research, rendering, and art teams.

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

Requirements:

• 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 one or more of these areas: 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
• Hands-on experience with interactive, action-conditioned, or real-time world models or video generation
• Experience developing custom CUDA or Triton kernels
• Experience with game engine development or real-time rendering pipelines
• Open-source contributions (such as Diffusers, FastVideo, vLLM, SGLang, or TensorRT-LLM), publications, or patents