lucidrains / ema-pytorch
A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
☆517Updated last month
Related projects ⓘ
Alternatives and complementary repositories for ema-pytorch
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆406Updated last month
- Neighborhood Attention Extension. Bringing attention to a neighborhood near you!☆366Updated this week
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆571Updated last week
- Unofficial Implementation of Consistency Models in pytorch☆251Updated last year
- Official Jax Implementation of MaskGIT☆449Updated 2 years ago
- MultiMAE: Multi-modal Multi-task Masked Autoencoders, ECCV 2022☆548Updated last year
- Implementation of Denoising Diffusion Probabilistic Models in PyTorch☆366Updated 2 years ago
- A series of tutorial notebooks on denoising diffusion probabilistic models in PyTorch☆647Updated 2 years ago
- Official implementation of Cold-Diffusion for different transformations in pytorch.☆1,013Updated 2 years ago
- PyTorch reimplementation of Diffusion Models☆504Updated last year
- High-fidelity performance metrics for generative models in PyTorch☆1,010Updated 9 months ago
- Pytorch implementation of MaskGIT: Masked Generative Image Transformer (https://arxiv.org/pdf/2202.04200.pdf)☆412Updated last year
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆265Updated 6 months ago
- Official implementation of Diffusion Autoencoders☆866Updated 2 months ago
- Implementation of Classifier Free Guidance in Pytorch, with emphasis on text conditioning, and flexibility to include multiple text embed…☆364Updated last month
- ☆238Updated last month
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆760Updated 4 months ago
- A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)☆645Updated last year
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆194Updated 9 months ago
- A PyTorch implementation of MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesis☆529Updated last year
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022☆1,054Updated 6 months ago
- An unofficial implementation of both ViT-VQGAN and RQ-VAE in Pytorch☆286Updated last year
- PyTorch - FID calculation with proper image resizing and quantization steps [CVPR 2022]☆971Updated 3 months ago
- A concise but complete implementation of CLIP with various experimental improvements from recent papers☆693Updated last year
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"☆360Updated last year
- Implementation of Bit Diffusion, Hinton's group's attempt at discrete denoising diffusion, in Pytorch☆333Updated last year
- Code for Fast Training of Diffusion Models with Masked Transformers☆373Updated 6 months ago
- Analyzing and Improving the Training Dynamics of Diffusion Models (EDM2)☆532Updated 5 months ago
- Official implementation of "DiffuseVAE: Efficient, Controllable and High-Fidelity Generation from Low-Dimensional Latents"☆353Updated 2 years ago
- The official implementation for Pseudo Numerical Methods for Diffusion Models on Manifolds (PNDM, PLMS | ICLR2022)☆332Updated last year