lucidrains / vector-quantize-pytorch
Vector (and Scalar) Quantization, in Pytorch
☆2,594Updated this week
Related projects ⓘ
Alternatives and complementary repositories for vector-quantize-pytorch
- PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆1,736Updated 3 months ago
- Release for Improved Denoising Diffusion Probabilistic Models☆3,257Updated 3 months ago
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,382Updated 7 months ago
- Denoising Diffusion Implicit Models☆1,451Updated 3 months ago
- Official code for "DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps" (Neurips 2022 Oral)☆1,556Updated 9 months ago
- TorchCFM: a Conditional Flow Matching library☆1,198Updated last month
- Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆1,494Updated last year
- A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)☆638Updated last year
- Denoising Diffusion Probabilistic Models☆3,832Updated last year
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆4,768Updated this week
- Diffusion model papers, survey, and taxonomy☆2,978Updated 3 months ago
- PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838☆984Updated last month
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆511Updated 2 weeks ago
- A series of tutorial notebooks on denoising diffusion probabilistic models in PyTorch☆645Updated last year
- Implementation of Denoising Diffusion Probabilistic Model in Pytorch☆8,308Updated last month
- Structured state space sequence models☆2,455Updated 3 months ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,092Updated last year
- Official implementation of Cold-Diffusion for different transformations in pytorch.☆1,010Updated 2 years ago
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,028Updated 4 months ago
- ☆916Updated last year
- TorchMultimodal is a PyTorch library for training state-of-the-art multimodal multi-task models at scale.☆1,460Updated this week
- This may be the simplest implement of DDPM. You can directly run Main.py to train the UNet on CIFAR-10 dataset and see the amazing proces…☆1,575Updated last year
- Official implementation of VQ-Diffusion☆894Updated 6 months ago
- Official Implementation of Rectified Flow (ICLR2023 Spotlight)☆929Updated 3 months ago
- Code for NeurIPS 2022 Paper, "Poisson Flow Generative Models" (PFGM)☆851Updated last year
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆565Updated last month
- High-fidelity performance metrics for generative models in PyTorch☆1,001Updated 9 months ago
- ☆6,241Updated 4 months ago
- Implementation of Video Diffusion Models, Jonathan Ho's new paper extending DDPMs to Video Generation - in Pytorch☆1,250Updated 6 months ago
- Implementation of Generating Diverse High-Fidelity Images with VQ-VAE-2 in PyTorch☆1,626Updated last year