lucidrains / ema-pytorchLinks
A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
☆611Updated 10 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.☆437Updated last year
- Official implementation of Cold-Diffusion for different transformations in pytorch.☆1,098Updated 2 years ago
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆757Updated 2 months ago
- Implementation of Denoising Diffusion Probabilistic Models in PyTorch☆389Updated 3 years ago
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆798Updated 3 months ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆288Updated last year
- Pytorch implementation of MaskGIT: Masked Generative Image Transformer (https://arxiv.org/pdf/2202.04200.pdf)☆450Updated 2 years ago
- A series of tutorial notebooks on denoising diffusion probabilistic models in PyTorch☆700Updated 2 years ago
- An unofficial implementation of both ViT-VQGAN and RQ-VAE in Pytorch☆316Updated 5 months ago
- High-fidelity performance metrics for generative models in PyTorch☆1,124Updated 2 months ago
- PyTorch reimplementation of Diffusion Models☆569Updated 2 years ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆771Updated 9 months ago
- Implementation of ConvMixer for "Patches Are All You Need? 🤷"☆1,077Updated 2 years ago
- A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)☆822Updated 2 years ago
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"☆383Updated 2 years ago
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022☆1,147Updated last year
- Unofficial Implementation of Consistency Models in pytorch☆259Updated 2 years ago
- A minimal yet resourceful implementation of diffusion models (along with pretrained models + synthetic images for nine datasets)☆293Updated last year
- An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow☆605Updated 11 months ago
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"☆820Updated 3 years ago
- Implementation of Bit Diffusion, Hinton's group's attempt at discrete denoising diffusion, in Pytorch☆355Updated last year
- Official Jax Implementation of MaskGIT☆535Updated 2 years ago
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆204Updated last year
- A PyTorch implementation of MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesis☆572Updated 2 years ago
- MultiMAE: Multi-modal Multi-task Masked Autoencoders, ECCV 2022☆594Updated 2 years ago
- Code for ICML 2023 paper, "PFGM++: Unlocking the Potential of Physics-Inspired Generative Models"☆376Updated 2 years ago
- ☆466Updated 2 years ago
- Official implementation of Diffusion Autoencoders☆942Updated last year
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆381Updated last month
- Code for NeurIPS 2022 Paper, "Poisson Flow Generative Models" (PFGM)☆867Updated 2 years ago