lucidrains / ema-pytorch
A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
☆552Updated last month
Alternatives and similar repositories for ema-pytorch:
Users that are interested in ema-pytorch are comparing it to the libraries listed below
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆417Updated 3 months ago
- PyTorch reimplementation of Diffusion Models☆514Updated 2 years ago
- Official implementation of Cold-Diffusion for different transformations in pytorch.☆1,044Updated 2 years ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆618Updated last month
- Official implementation of Diffusion Autoencoders☆887Updated 4 months ago
- Implementation of Denoising Diffusion Probabilistic Models in PyTorch☆372Updated 2 years ago
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆614Updated 2 months ago
- Code for Fast Training of Diffusion Models with Masked Transformers☆386Updated 8 months ago
- Neighborhood Attention Extension. Bringing attention to a neighborhood near you!☆396Updated 3 weeks ago
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆772Updated 6 months ago
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"☆370Updated last year
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,495Updated 10 months ago
- Pytorch implementation of MaskGIT: Masked Generative Image Transformer (https://arxiv.org/pdf/2202.04200.pdf)☆421Updated last year
- Official Jax Implementation of MaskGIT☆479Updated 2 years ago
- A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)☆688Updated 2 years ago
- Tackling the Generative Learning Trilemma with Denoising Diffusion GANs https://arxiv.org/abs/2112.07804☆719Updated 2 years ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆271Updated 8 months ago
- Unofficial Implementation of Consistency Models in pytorch☆255Updated last year
- Denoising Diffusion Implicit Models☆1,537Updated 6 months ago
- Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)☆544Updated 9 months ago
- Official Implementation of Rectified Flow (ICLR2023 Spotlight)☆1,067Updated 6 months ago
- A PyTorch implementation of MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesis☆548Updated last year
- Implementation of Classifier Free Guidance in Pytorch, with emphasis on text conditioning, and flexibility to include multiple text embed…☆395Updated 3 months ago
- An unofficial implementation of both ViT-VQGAN and RQ-VAE in Pytorch☆295Updated last year
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆197Updated 11 months ago
- A minimal yet resourceful implementation of diffusion models (along with pretrained models + synthetic images for nine datasets)☆267Updated 4 months ago
- Implementation of Bit Diffusion, Hinton's group's attempt at discrete denoising diffusion, in Pytorch☆341Updated last year
- A concise but complete implementation of CLIP with various experimental improvements from recent papers☆705Updated last year
- Official Pytorch Implementation of Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think (ICL…☆815Updated this week
- ☆304Updated 2 years ago