AI Software Engineer

Nvidia

Confirmed live yesterday High trust
Remote

Quick summary

Work type
Remote
Location
Santa Clara, CA
Salary
$224,000–$356,500 / yr
Employment
Full-time
Posted
5 days ago
Freshness
Confirmed live yesterday

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $192k
This role $290k
$118k most similar roles pay here $382k

This role pays more than 97% of similar roles. Most pay $167,375–$217,525 — the shaded band above. At the midpoint, this role pays about $290k versus about $192k for comparable roles.

Based on 240 similar postings.

Employer

About Nvidia

Nvidia is a leading designer of graphics processing units (GPUs) and system-on-chip units, powering gaming, professional visualization, data centers, and artificial intelligence workloads. Industry: Semiconductors & AI Computing

Nvidia currently has 1371 open roles on FindRole.

Listed pay typically runs $184,000–$287,500 across 1096 roles with salary data.

Most-posted roles

View all roles at Nvidia

At a glance

TL;DR · AI Software Engineer

AI Software Engineer, Lightspeed Studios joins the engineering team to develop and ship software for unannounced projects that integrate world models and video diffusion models into real-time gaming experiences. The role involves taking products from prototype to release by training, fine-tuning, and evaluating models while performing data curation and optimizing inference for latency and quality using distillation and quantization. Key responsibilities include profiling bottlenecks across the stack, developing custom kernels, and integrating models into Unreal Engine, Unity, or custom engines. Candidates must possess expert-level C++ and Python skills along with experience in PyTorch, CUDA, TensorRT, TensorRT-LLM, or Triton. The work focuses on solving technical challenges in neural rendering, transformer models, and flow-matching to create interactive gaming content using advanced AI technologies like generative tools for textures and autonomous game characters.

What does a AI Software Engineer earn?

Median $196000 from 30 postings across 13 companies.

See salary data

What you'll do

  • Develop and ship software for upcoming projects integrating world models and video diffusion into real-time gaming.
  • Train, fine-tune, and evaluate deep learning models using curated data tailored to game-specific content.
  • Optimize model inference for latency, token throughput, and quality using distillation, quantization, and reduced-step sampling.
  • Profile and resolve performance bottlenecks across the stack from model architecture down to GPU kernels.
  • Integrate AI models into Unreal Engine, Unity, and custom engines to run efficiently alongside rendering pipelines.
  • Lead technical decision-making and mentor other engineers within the studio.

What we're looking for

  • 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 proficiency in 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/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 teams.
  • Experience with interactive world models, custom CUDA/Triton kernels, game engine development, or open-source contributions (preferred).

More like this

Similar roles

Senior Software Engineer

Nvidia

Santa Clara, CA 44 days ago $224,000–$356,500
C++ Python Linux Computer Vision Deep Learning PyTorch CUDA TensorRT Distributed Systems Microservices Containers GPU Acceleration Multi-threading Asynchronous I/O Omniverse Unreal Engine Unity
10+ yrs exp

AI Software Engineer

Allstate

Remote (IL) 19 days ago $110,000–$150,000
Generative AI LLM LangChain Python Java Spring Boot Node.js FastAPI RESTful APIs Microservices Docker Kubernetes CI/CD Microsoft Azure AWS GCP RAG Vector Databases React Angular
4+ yrs exp Remote