Alternatives and similar repositories for pytorch_ema

Users that are interested in pytorch_ema are comparing it to the libraries listed below

• Tony-Y / pytorch_warmup
  Learning Rate Warmup in PyTorch
  ☆412Updated 3 months ago
• lucidrains / ema-pytorch
  A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
  ☆609Updated 9 months ago
• katsura-jp / pytorch-cosine-annealing-with-warmup
  ☆466Updated 2 years ago
• kakaobrain / torchlars
  A LARS implementation in PyTorch
  ☆352Updated 5 years ago
• wzlxjtu / PositionalEncoding2D
  A PyTorch implementation of the 1d and 2d Sinusoidal positional encoding/embedding.
  ☆257Updated 4 years ago
• locuslab / convmixer
  Implementation of ConvMixer for "Patches Are All You Need? 🤷"
  ☆1,077Updated 2 years ago
• xxxnell / how-do-vits-work
  (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"
  ☆820Updated 3 years ago
• imbue-ai / self_supervised
  A Pytorch-Lightning implementation of self-supervised algorithms
  ☆544Updated 3 years ago
• DeMoriarty / fast_pytorch_kmeans
  This is a pytorch implementation of k-means clustering algorithm
  ☆325Updated 6 months ago
• lessw2020 / Ranger21
  Ranger deep learning optimizer rewrite to use newest components
  ☆336Updated last year
• lukemelas / do-you-even-need-attention
  Is the attention layer even necessary? (https://arxiv.org/abs/2105.02723)
  ☆486Updated 4 years ago
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆1,045Updated 2 months ago
• sail-sg / Adan
  Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
  ☆798Updated 3 months ago
• fkodom / fft-conv-pytorch
  Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.
  ☆507Updated 2 years ago
• lucidrains / axial-attention
  Implementation of Axial attention - attending to multi-dimensional data efficiently
  ☆386Updated 4 years ago
• facebookresearch / msn
  Masked Siamese Networks for Label-Efficient Learning (https://arxiv.org/abs/2204.07141)
  ☆458Updated 3 years ago
• lucidrains / transformer-in-transformer
  Implementation of Transformer in Transformer, pixel level attention paired with patch level attention for image classification, in Pytorc…
  ☆309Updated 3 years ago
• facebookresearch / barlowtwins
  PyTorch implementation of Barlow Twins.
  ☆994Updated 3 years ago
• lucidrains / linformer
  Implementation of Linformer for Pytorch
  ☆300Updated last year
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆798Updated last year
• assafshocher / ResizeRight
  The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).
  ☆564Updated 2 years ago
• microsoft / esvit
  EsViT: Efficient self-supervised Vision Transformers
  ☆412Updated 2 years ago
• AndrewAtanov / simclr-pytorch
  PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results
  ☆208Updated last year
• nocotan / pytorch-lightning-gans
  Collection of PyTorch Lightning implementations of Generative Adversarial Network varieties presented in research papers.
  ☆169Updated 6 months ago
• google-research / sam
  ☆605Updated last month
• ildoonet / pytorch-gradual-warmup-lr
  Gradually-Warmup Learning Rate Scheduler for PyTorch
  ☆992Updated 11 months ago
• rishikksh20 / MLP-Mixer-pytorch
  Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision
  ☆218Updated 4 years ago
• teddykoker / torchsort
  Fast, differentiable sorting and ranking in PyTorch
  ☆834Updated 3 months ago
• vballoli / nfnets-pytorch
  NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/
  ☆349Updated last year
• AdeelH / pytorch-multi-class-focal-loss
  An (unofficial) implementation of Focal Loss, as described in the RetinaNet paper, generalized to the multi-class case.
  ☆238Updated last year