Running large language models like ChatGPT/GPT-3/OPT-175B on a single GPU. Up to 100x faster than other offloading systems.

Compute FID scores with PyTorch.

A latent text-to-image diffusion model

Official PyTorch implementation of StyleGAN3

Repository for the paper "Very Deep VAEs Generalize Autoregressive Models and Can Outperform Them on Images"

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)

Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"

Ladder Variational Autoencoders (LVAE) in PyTorch

Basic framework for experiment management in PyTorch.

A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.

PyTorch deep learning projects made easy.

Official PyTorch BIVA implementation (BIVA: A Very Deep Hierarchy of Latent Variables for Generative Modeling)

Code for Neural Arithmetic Units (ICLR) and Measuring Arithmetic Extrapolation Performance (SEDL|NeurIPS)

Multimodal Mixture-of-Experts VAE

Latent Normalizing Flows for Many-to-Many Cross Domain Mappings (ICLR 2020)

Code examples in pyTorch and Tensorflow for CS230

BIVA: A Very Deep Hierarchy of Latent Variables forGenerative Modeling

A PyTorch re-implementation of "Doubly Reparameterized Gradient Estimators for Monte Carlo Objectives"

A PyTorch implementation of "Multimodal Generative Models for Scalable Weakly-Supervised Learning" (https://arxiv.org/abs/1802.05335)