Nvidia

AI Software Engineer, Lightspeed Studios

Nvidia
Apply
2 days ago
Remote, United States or Santa Clara, CA, USASenior
H1B sponsor

Base Salary

$224k - $431k/yr

Responsibilities

  • Build and ship software that brings world models and video diffusion models to real-time gaming experiences.
  • Train, fine-tune, evaluate, and adapt deep learning models using curated, game-specific data and controls.
  • Optimize model inference for latency, token throughput, and quality through distillation, quantization, and reduced-step sampling.
  • Profile and remove performance bottlenecks across model architectures, GPU kernels, RTX devices, and cloud infrastructure.
  • Integrate AI models into Unreal Engine, Unity, and custom game engines alongside rendering systems.
  • Lead technical decisions, mentor engineers, and collaborate with NVIDIA Research, rendering, and art teams.

Requirements

  • A 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 experience shipping AI, deep learning, or machine learning systems.
  • Expert-level C++ and Python skills.
  • Hands-on experience training and fine-tuning deep learning models with PyTorch or a similar framework, including distributed training.
  • Experience with 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 record of turning complex prototypes into shipped products and communicating across research, engineering, and art.
  • Preferred experience with interactive, action-conditioned, or real-time world models or video generation.
  • Preferred experience developing custom CUDA or Triton kernels and working with game-engine development or real-time rendering pipelines.
  • Open-source contributions to projects such as Diffusers, FastVideo, vLLM, SGLang, or TensorRT-LLM, publications, or patents are advantageous.

Benefits

  • Eligible for equity and benefits.
  • Base salary is determined by location, experience, and comparable employee pay.
  • Applications will be accepted at least until October 2, 2026.

Tech Stack

C++PythonPyTorchUnityUnreal Engine

Categories

Nvidia

About Nvidia

10,000+ employees

Nvidia designs and sells GPUs and accelerated computing platforms for data centers, AI/ML, graphics, gaming, and automotive, monetizing through hardware, software platforms (CUDA, AI frameworks), and systems like DGX and networking. Customers include cloud providers, enterprises, researchers, and OEMs. Founded in 1993 and headquartered in Santa Clara, it is a public company traded on NASDAQ under NVDA.

Contact me