AI Software Engineer

This is a fully remote position, open to applicants in California.

📋 Description

• Develop and deliver software for forthcoming, unannounced initiatives that incorporate world models and video diffusion models into real-time gaming experiences.

• Train, refine, and assess models, which includes data curation and tailoring models to game-specific elements and controls.

• Enhance models and inference for latency, token throughput, and quality by employing distillation, quantization, and reduced-step sampling on RTX devices and in cloud environments.

• Identify and eliminate bottlenecks throughout the stack, ranging from model architecture to GPU kernels.

• Seamlessly integrate models into Unreal Engine, Unity, and proprietary engines, ensuring efficient operation alongside rendering tasks.

• Guide technical choices, mentor fellow engineers, and collaborate effectively with NVIDIA Research, rendering, and art teams.

⛳️ Requirements

• Bachelor's, Master's, or PhD in Computer Science, Electrical Engineering, or a related discipline, or equivalent experience.

• Over 12 years of experience in software engineering, with significant hands-on expertise in deploying AI, deep learning, or machine learning systems.

• Proficient in C++ and Python at an expert level.

• Practical experience in training and fine-tuning deep learning models using PyTorch or a comparable framework, including distributed training capabilities.

• Familiarity with video generation, world models, diffusion or flow-matching models, transformers, or neural rendering.

• Experience in optimizing GPU inference utilizing CUDA, TensorRT, TensorRT-LLM, or Triton.

• Proven history of transforming intricate prototypes into commercially released products.

• Strong communication skills across research, engineering, and artistic domains.

• Practical experience with interactive, action-conditioned, or real-time world models or video generation is a plus.

• Experience in developing custom CUDA or Triton kernels is a plus.

• Familiarity with game engine development or real-time rendering pipelines is a plus.

• Contributions to open-source projects, publications, or patents are advantageous.

• Resume and cover letter are required.

🏝️ Benefits

• Equity

• Benefits