Alternatives and similar repositories for ema-pytorch:

Users that are interested in ema-pytorch are comparing it to the libraries listed below

• fadel / pytorch_ema
  Tiny PyTorch library for maintaining a moving average of a collection of parameters.
  ☆427Updated 5 months ago
• arpitbansal297 / Cold-Diffusion-Models
  Official implementation of Cold-Diffusion for different transformations in pytorch.
  ☆1,065Updated 2 years ago
• dome272 / MaskGIT-pytorch
  Pytorch implementation of MaskGIT: Masked Generative Image Transformer (https://arxiv.org/pdf/2202.04200.pdf)
  ☆430Updated last year
• google-research / maskgit
  Official Jax Implementation of MaskGIT
  ☆499Updated 2 years ago
• pesser / pytorch_diffusion
  PyTorch reimplementation of Diffusion Models
  ☆534Updated 2 years ago
• sail-sg / Adan
  Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
  ☆783Updated 8 months ago
• LTH14 / mage
  A PyTorch implementation of MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesis
  ☆555Updated 2 years ago
• lucidrains / memory-efficient-attention-pytorch
  Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"
  ☆374Updated last year
• lucidrains / rotary-embedding-torch
  Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch
  ☆653Updated 4 months ago
• acids-ircam / diffusion_models
  A series of tutorial notebooks on denoising diffusion probabilistic models in PyTorch
  ☆672Updated 2 years ago
• thuanz123 / enhancing-transformers
  An unofficial implementation of both ViT-VQGAN and RQ-VAE in Pytorch
  ☆299Updated last year
• NVlabs / edm2
  EDM2 and Autoguidance -- Official PyTorch implementation
  ☆667Updated 3 months ago
• MishaLaskin / vqvae
  A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)
  ☆724Updated 2 years ago
• EPFL-VILAB / MultiMAE
  MultiMAE: Multi-modal Multi-task Masked Autoencoders, ECCV 2022
  ☆570Updated 2 years ago
• toshas / torch-fidelity
  High-fidelity performance metrics for generative models in PyTorch
  ☆1,056Updated last year
• NVlabs / edm
  Elucidating the Design Space of Diffusion-Based Generative Models (EDM)
  ☆1,588Updated last year
• assafshocher / ResizeRight
  The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).
  ☆557Updated last year
• google-research / vdm
  ☆310Updated 2 years ago
• NVlabs / denoising-diffusion-gan
  Tackling the Generative Learning Trilemma with Denoising Diffusion GANs https://arxiv.org/abs/2112.07804
  ☆726Updated 2 years ago
• lucidrains / bit-diffusion
  Implementation of Bit Diffusion, Hinton's group's attempt at discrete denoising diffusion, in Pytorch
  ☆342Updated last year
• lucidrains / x-unet
  Implementation of a U-net complete with efficient attention as well as the latest research findings
  ☆275Updated 10 months ago
• phizaz / diffae
  Official implementation of Diffusion Autoencoders
  ☆906Updated 6 months ago
• lucidrains / x-clip
  A concise but complete implementation of CLIP with various experimental improvements from recent papers
  ☆708Updated last year
• sail-sg / MDT
  Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)
  ☆554Updated 11 months ago
• locuslab / convmixer
  Implementation of ConvMixer for "Patches Are All You Need? 🤷"
  ☆1,071Updated 2 years ago
• SHI-Labs / NATTEN
  Neighborhood Attention Extension. Bringing attention to a neighborhood near you!
  ☆430Updated 2 weeks ago
• cientgu / VQ-Diffusion
  ☆455Updated 2 years ago
• rosinality / denoising-diffusion-pytorch
  Implementation of Denoising Diffusion Probabilistic Models in PyTorch
  ☆376Updated 2 years ago
• GaParmar / clean-fid
  PyTorch - FID calculation with proper image resizing and quantization steps [CVPR 2022]
  ☆1,039Updated 8 months ago
• cloneofsimo / consistency_models
  Unofficial Implementation of Consistency Models in pytorch
  ☆253Updated 2 years ago