lucidrains / vector-quantize-pytorch
Vector (and Scalar) Quantization, in Pytorch
☆2,639Updated last week
Related projects ⓘ
Alternatives and complementary repositories for vector-quantize-pytorch
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,400Updated 8 months ago
- PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆1,747Updated 4 months ago
- Diffusion model papers, survey, and taxonomy☆2,994Updated 3 months ago
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,041Updated 5 months ago
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆4,793Updated this week
- Release for Improved Denoising Diffusion Probabilistic Models☆3,279Updated 4 months ago
- A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)☆645Updated last year
- Structured state space sequence models☆2,470Updated 4 months ago
- Denoising Diffusion Implicit Models☆1,463Updated 3 months ago
- Official implementation of Cold-Diffusion for different transformations in pytorch.☆1,013Updated 2 years ago
- ☆920Updated last year
- Official code for "DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps" (Neurips 2022 Oral)☆1,564Updated 9 months ago
- TorchCFM: a Conditional Flow Matching library☆1,238Updated this week
- TorchMultimodal is a PyTorch library for training state-of-the-art multimodal multi-task models at scale.☆1,474Updated this week
- Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆1,503Updated last year
- Official implementation of VQ-Diffusion☆899Updated 7 months ago
- Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"☆6,369Updated 5 months ago
- Implementation of Denoising Diffusion Probabilistic Model in Pytorch☆8,373Updated last month
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆517Updated last month
- Denoising Diffusion Probabilistic Models☆3,867Updated last year
- A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.☆2,328Updated last month
- PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838☆1,011Updated last month
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆571Updated last week
- Implementation of Generating Diverse High-Fidelity Images with VQ-VAE-2 in PyTorch☆1,635Updated last year
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆698Updated 6 months ago
- Official Implementation of Rectified Flow (ICLR2023 Spotlight)☆954Updated 4 months ago
- ☆6,275Updated 4 months ago
- A series of tutorial notebooks on denoising diffusion probabilistic models in PyTorch☆647Updated 2 years ago
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆760Updated 4 months ago
- An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow☆546Updated 3 weeks ago