Datasets, Transforms and Models specific to Computer Vision

High-Resolution Image Synthesis with Latent Diffusion Models

8-bit CUDA functions for PyTorch

Learning to See by Looking at Noise

Development repository for the Triton language and compiler

Instant neural graphics primitives: lightning fast NeRF and more

Official PyTorch implementation of StyleGAN3

Code release for DriveGAN (CVPR 2021)

Learning to Simulate Dynamic Environments with GameGAN (CVPR 2020)

NVIDIA PyTorch GAN library with distributed and mixed precision support

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch

A latent text-to-image diffusion model