lucidrains / vector-quantize-pytorchLinks
Vector (and Scalar) Quantization, in Pytorch
☆3,857Updated 2 weeks ago
Alternatives and similar repositories for vector-quantize-pytorch
Users that are interested in vector-quantize-pytorch are comparing it to the libraries listed below
Sorting:
- A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)☆875Updated 3 years ago
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆5,800Updated this week
- TorchCFM: a Conditional Flow Matching library☆2,282Updated 2 months ago
- PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆2,079Updated last year
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆4,107Updated last month
- PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838☆1,857Updated last year
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,901Updated last year
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆802Updated last week
- Denoising Diffusion Implicit Models☆1,782Updated last year
- Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆1,799Updated 3 years ago
- Diffusion model papers, survey, and taxonomy☆3,321Updated 4 months ago
- Official Implementation of Rectified Flow (ICLR2023 Spotlight)☆1,543Updated last year
- Release for Improved Denoising Diffusion Probabilistic Models☆3,782Updated last year
- Denoising Diffusion Probabilistic Models☆5,035Updated 2 years ago
- Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI☆1,322Updated last week
- Official code for "DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps" (Neurips 2022 Oral)☆1,815Updated 2 years ago
- Conditional diffusion model to generate MNIST. Minimal script. Based on 'Classifier-Free Diffusion Guidance'.☆825Updated 2 years ago
- Muon is an optimizer for hidden layers in neural networks☆2,267Updated 3 weeks ago
- Pytorch implementation of Diffusion Models (https://arxiv.org/pdf/2006.11239.pdf)☆1,426Updated 2 years ago
- Implementation of Generating Diverse High-Fidelity Images with VQ-VAE-2 in PyTorch☆1,795Updated 2 years ago
- PyTorch Re-Implementation of "The Sparsely-Gated Mixture-of-Experts Layer" by Noam Shazeer et al. https://arxiv.org/abs/1701.06538☆1,225Updated last year
- 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…☆2,140Updated 2 years ago
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,182Updated last year
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆640Updated last month
- Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.☆3,343Updated 8 months ago
- Pytorch implementation of VQGAN (Taming Transformers for High-Resolution Image Synthesis) (https://arxiv.org/pdf/2012.09841.pdf)☆542Updated last year
- A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.☆2,981Updated this week
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆827Updated last year
- PyTorch implementation of FractalGen https://arxiv.org/abs/2502.17437☆1,220Updated 11 months ago
- A Pytorch implementation of Sparsely-Gated Mixture of Experts, for massively increasing the parameter count of language models☆846Updated 2 years ago