lucidrains / ema-pytorchLinks
A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
☆592Updated 6 months ago
Alternatives and similar repositories for ema-pytorch
Users that are interested in ema-pytorch are comparing it to the libraries listed below
Sorting:
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆430Updated 8 months ago
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆792Updated 2 weeks ago
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆694Updated 6 months ago
- Pytorch implementation of MaskGIT: Masked Generative Image Transformer (https://arxiv.org/pdf/2202.04200.pdf)☆444Updated last year
- Official implementation of Cold-Diffusion for different transformations in pytorch.☆1,080Updated 2 years ago
- High-fidelity performance metrics for generative models in PyTorch☆1,094Updated last year
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"☆379Updated last year
- A PyTorch implementation of MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesis☆565Updated 2 years ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆729Updated 6 months ago
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"☆817Updated 2 years ago
- A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)☆768Updated 2 years ago
- ☆320Updated 2 years ago
- Implementation of Denoising Diffusion Probabilistic Models in PyTorch☆384Updated 3 years ago
- Official Jax Implementation of MaskGIT☆519Updated 2 years ago
- Official implementation of Diffusion Autoencoders☆928Updated 9 months ago
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,697Updated last year
- Unofficial Implementation of Consistency Models in pytorch☆257Updated 2 years ago
- A minimal yet resourceful implementation of diffusion models (along with pretrained models + synthetic images for nine datasets)☆286Updated 9 months ago
- Implementation of Bit Diffusion, Hinton's group's attempt at discrete denoising diffusion, in Pytorch☆350Updated last year
- Neighborhood Attention Extension. Bringing attention to a neighborhood near you!☆528Updated this week
- PyTorch reimplementation of Diffusion Models☆556Updated 2 years ago
- Official implementation of VQ-Diffusion☆945Updated last year
- A series of tutorial notebooks on denoising diffusion probabilistic models in PyTorch☆694Updated 2 years ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆282Updated last year
- ☆464Updated 2 years ago
- [ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation☆495Updated 7 months ago
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022☆1,122Updated last year
- Tackling the Generative Learning Trilemma with Denoising Diffusion GANs https://arxiv.org/abs/2112.07804☆742Updated 2 years ago
- Implementation of ConvMixer for "Patches Are All You Need? 🤷"☆1,074Updated 2 years ago
- Official implementation of "DiffuseVAE: Efficient, Controllable and High-Fidelity Generation from Low-Dimensional Latents"☆369Updated 2 years ago