lucidrains / ema-pytorchLinks
A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
☆590Updated 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.☆429Updated 8 months ago
- Pytorch implementation of MaskGIT: Masked Generative Image Transformer (https://arxiv.org/pdf/2202.04200.pdf)☆444Updated last year
- EDM2 and Autoguidance -- Official PyTorch implementation☆719Updated 5 months ago
- Official implementation of Cold-Diffusion for different transformations in pytorch.☆1,075Updated 2 years ago
- Official Jax Implementation of MaskGIT☆513Updated 2 years ago
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆791Updated 11 months ago
- Implementation of Denoising Diffusion Probabilistic Models in PyTorch☆381Updated 2 years ago
- An unofficial implementation of both ViT-VQGAN and RQ-VAE in Pytorch☆305Updated last month
- A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)☆760Updated 2 years ago
- Unofficial Implementation of Consistency Models in pytorch☆256Updated 2 years ago
- A PyTorch implementation of MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesis☆564Updated 2 years ago
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆685Updated 6 months ago
- Official implementation of Diffusion Autoencoders☆923Updated 8 months ago
- Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)☆566Updated last year
- High-fidelity performance metrics for generative models in PyTorch☆1,080Updated last year
- [ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation☆492Updated 7 months ago
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022☆1,118Updated last year
- Implementation of Bit Diffusion, Hinton's group's attempt at discrete denoising diffusion, in Pytorch☆346Updated last year
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆853Updated last year
- PyTorch reimplementation of Diffusion Models☆549Updated 2 years ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆282Updated last year
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,668Updated last year
- PyTorch implementation of RCG https://arxiv.org/abs/2312.03701☆915Updated 8 months ago
- Official Pytorch Implementation of the paper: Wavelet Diffusion Models are fast and scalable Image Generators (CVPR'23)☆412Updated 10 months ago
- Code for Fast Training of Diffusion Models with Masked Transformers☆403Updated last year
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"☆378Updated last year
- Code for NeurIPS 2022 Paper, "Poisson Flow Generative Models" (PFGM)☆860Updated last year
- A series of tutorial notebooks on denoising diffusion probabilistic models in PyTorch☆688Updated 2 years ago
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆473Updated last year
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆203Updated last year