karpathy / deep-vector-quantizationLinks
VQVAEs, GumbelSoftmaxes and friends
☆624Updated 4 years ago
Alternatives and similar repositories for deep-vector-quantization
Users that are interested in deep-vector-quantization are comparing it to the libraries listed below
Sorting:
- Normalizing flows in PyTorch. Current intended use is education not production.☆905Updated 5 years ago
- Repository for the paper "Very Deep VAEs Generalize Autoregressive Models and Can Outperform Them on Images"☆449Updated 2 years ago
- The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)☆1,082Updated 3 years ago
- ☆328Updated 3 years ago
- Easy generative modeling in PyTorch☆438Updated 2 years ago
- Minimalist implementation of VQ-VAE in Pytorch☆556Updated 4 years ago
- PyTorch implementation of VQ-VAE by Aäron van den Oord et al.☆598Updated 6 years ago
- A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)☆858Updated 3 years ago
- Official implementation of VQ-Diffusion☆971Updated last year
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆501Updated last year
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆629Updated last year
- An implementation of local windowed attention for language modeling☆491Updated 5 months ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆818Updated last year
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,186Updated 2 years ago
- PyTorch Implementation of OpenAI's Image GPT☆260Updated 2 years ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆443Updated last year
- Noise Conditional Score Networks (NeurIPS 2019, Oral)☆757Updated last year
- Implementation of Slot Attention from GoogleAI☆476Updated last year
- Long Range Arena for Benchmarking Efficient Transformers☆771Updated 2 years ago
- ☆387Updated 2 years ago
- A series of tutorial notebooks on denoising diffusion probabilistic models in PyTorch☆714Updated 3 years ago
- Fully featured implementation of Routing Transformer☆299Updated 4 years ago
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆783Updated 4 months ago
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"☆388Updated 2 years ago
- Vector Quantized VAEs - PyTorch Implementation☆943Updated 2 years ago
- Simple, extendable, easy to understand Glow implementation in PyTorch☆387Updated 3 years ago
- Official Pytorch and JAX implementation of "Efficient-VDVAE: Less is more"☆196Updated 3 years ago
- The Official PyTorch Implementation of "LSGM: Score-based Generative Modeling in Latent Space" (NeurIPS 2021)☆370Updated 4 years ago
- An implementation of Performer, a linear attention-based transformer, in Pytorch☆1,169Updated 3 years ago
- Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda☆230Updated 2 years ago