lucidrains / vector-quantize-pytorch
Vector (and Scalar) Quantization, in Pytorch
☆2,922Updated last week
Alternatives and similar repositories for vector-quantize-pytorch:
Users that are interested in vector-quantize-pytorch are comparing it to the libraries listed below
- A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)☆699Updated 2 years ago
- PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆1,836Updated 7 months ago
- Release for Improved Denoising Diffusion Probabilistic Models☆3,429Updated 7 months ago
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,528Updated 11 months ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆560Updated 2 months ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,120Updated last year
- Denoising Diffusion Implicit Models☆1,555Updated 6 months ago
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆5,080Updated this week
- PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838☆1,285Updated 4 months ago
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆639Updated 2 months ago
- Implementation of Denoising Diffusion Probabilistic Model in Pytorch☆8,863Updated 4 months ago
- Official implementation of VQ-Diffusion☆914Updated 10 months ago
- Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆1,574Updated 2 years ago
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,095Updated 2 months ago
- Official code for "DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps" (Neurips 2022 Oral)☆1,619Updated last year
- TorchCFM: a Conditional Flow Matching library☆1,483Updated 3 weeks ago
- Official Implementation of Rectified Flow (ICLR2023 Spotlight)☆1,096Updated 7 months ago
- Official implementation of Cold-Diffusion for different transformations in pytorch.☆1,053Updated 2 years ago
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆774Updated 7 months ago
- TorchMultimodal is a PyTorch library for training state-of-the-art multimodal multi-task models at scale.☆1,541Updated this week
- Structured state space sequence models☆2,553Updated 7 months ago
- Pytorch library for fast transformer implementations☆1,679Updated last year
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆1,986Updated last month
- Denoising Diffusion Probabilistic Models☆4,141Updated last year
- An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow☆566Updated 3 months ago
- A collection of literature after or concurrent with Masked Autoencoder (MAE) (Kaiming He el al.).☆805Updated 7 months ago
- Implementation of Video Diffusion Models, Jonathan Ho's new paper extending DDPMs to Video Generation - in Pytorch☆1,282Updated 9 months ago
- PyTorch Re-Implementation of "The Sparsely-Gated Mixture-of-Experts Layer" by Noam Shazeer et al. https://arxiv.org/abs/1701.06538☆1,043Updated 10 months ago
- Implementation of Generating Diverse High-Fidelity Images with VQ-VAE-2 in PyTorch☆1,686Updated 2 years ago
- High-fidelity performance metrics for generative models in PyTorch☆1,042Updated last year