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 concise but complete full-attention transformer with a set of promising experimental features from various papers

Vector (and Scalar) Quantization, in Pytorch

Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch