karpathy / deep-vector-quantizationLinks
VQVAEs, GumbelSoftmaxes and friends
☆582Updated 3 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:
- Repository for the paper "Very Deep VAEs Generalize Autoregressive Models and Can Outperform Them on Images"☆446Updated 2 years ago
- The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)☆1,062Updated 2 years ago
- PyTorch implementation of VQ-VAE by Aäron van den Oord et al.☆587Updated 5 years ago
- Minimalist implementation of VQ-VAE in Pytorch☆547Updated 4 years ago
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆487Updated last year
- Easy generative modeling in PyTorch☆440Updated last year
- ☆321Updated 2 years ago
- Normalizing flows in PyTorch. Current intended use is education not production.☆876Updated 5 years ago
- A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)☆799Updated 2 years ago
- Noise Conditional Score Networks (NeurIPS 2019, Oral)☆738Updated last year
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,165Updated 2 years ago
- Official implementation of VQ-Diffusion☆954Updated last year
- ☆379Updated last year
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆794Updated last year
- A series of tutorial notebooks on denoising diffusion probabilistic models in PyTorch☆696Updated 2 years ago
- Vector Quantized VAEs - PyTorch Implementation☆925Updated 2 years ago
- PyTorch Implementation of OpenAI's Image GPT☆258Updated last year
- Implementation of Slot Attention from GoogleAI☆452Updated last year
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆601Updated 8 months ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆435Updated 10 months ago
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆479Updated 3 years ago
- Official codebase for Pretrained Transformers as Universal Computation Engines.☆248Updated 3 years ago
- ☆812Updated 4 months ago
- Long Range Arena for Benchmarking Efficient Transformers☆762Updated last year
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆251Updated 2 years ago
- Fully featured implementation of Routing Transformer☆297Updated 3 years ago
- Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory☆438Updated 11 months ago
- A library to inspect and extract intermediate layers of PyTorch models.☆473Updated 3 years ago
- An implementation of local windowed attention for language modeling☆472Updated last month
- Simple, extendable, easy to understand Glow implementation in PyTorch☆385Updated 3 years ago