lucidrains / ema-pytorch
A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
☆571Updated 4 months ago
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.☆426Updated 6 months ago
- A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)☆737Updated 2 years ago
- A PyTorch implementation of MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesis☆557Updated 2 years ago
- Official Jax Implementation of MaskGIT☆507Updated 2 years ago
- A series of tutorial notebooks on denoising diffusion probabilistic models in PyTorch☆676Updated 2 years ago
- Official implementation of Cold-Diffusion for different transformations in pytorch.☆1,070Updated 2 years ago
- Official implementation of Diffusion Autoencoders☆913Updated 7 months ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆278Updated 11 months ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆693Updated 4 months ago
- Implementation of Denoising Diffusion Probabilistic Models in PyTorch☆379Updated 2 years ago
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"☆377Updated last year
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆666Updated 4 months ago
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,623Updated last year
- PyTorch reimplementation of Diffusion Models☆540Updated 2 years ago
- Pytorch implementation of MaskGIT: Masked Generative Image Transformer (https://arxiv.org/pdf/2202.04200.pdf)☆436Updated last year
- MultiMAE: Multi-modal Multi-task Masked Autoencoders, ECCV 2022☆572Updated 2 years ago
- Unofficial Implementation of Consistency Models in pytorch☆254Updated 2 years ago
- High-fidelity performance metrics for generative models in PyTorch☆1,067Updated last year
- [ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation☆492Updated 5 months ago
- [ECCV 2024] Official PyTorch implementation of RoPE-ViT "Rotary Position Embedding for Vision Transformer"☆315Updated 4 months ago
- The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).☆558Updated last year
- Tackling the Generative Learning Trilemma with Denoising Diffusion GANs https://arxiv.org/abs/2112.07804☆732Updated 2 years ago
- A collection of literature after or concurrent with Masked Autoencoder (MAE) (Kaiming He el al.).☆820Updated 9 months ago
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆789Updated 9 months ago
- ☆312Updated 2 years ago
- An unofficial implementation of both ViT-VQGAN and RQ-VAE in Pytorch☆300Updated 2 weeks ago
- Neighborhood Attention Extension. Bringing attention to a neighborhood near you!☆443Updated last month
- ☆461Updated 2 years ago
- Official implementation of "DiffuseVAE: Efficient, Controllable and High-Fidelity Generation from Low-Dimensional Latents"☆366Updated 2 years ago
- PyTorch - FID calculation with proper image resizing and quantization steps [CVPR 2022]☆1,055Updated 9 months ago