Alternatives and similar repositories for axial-attention

Users that are interested in axial-attention are comparing it to the libraries listed below

• cmsflash / efficient-attention
  An implementation of the efficient attention module.
  ☆319Updated 4 years ago
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆1,029Updated last week
• rishikksh20 / MLP-Mixer-pytorch
  Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision
  ☆218Updated 4 years ago
• locuslab / convmixer
  Implementation of ConvMixer for "Patches Are All You Need? 🤷"
  ☆1,074Updated 2 years ago
• csrhddlam / axial-deeplab
  This is a PyTorch re-implementation of Axial-DeepLab (ECCV 2020 Spotlight)
  ☆456Updated 4 years ago
• lucidrains / transformer-in-transformer
  Implementation of Transformer in Transformer, pixel level attention paired with patch level attention for image classification, in Pytorc…
  ☆305Updated 3 years ago
• fadel / pytorch_ema
  Tiny PyTorch library for maintaining a moving average of a collection of parameters.
  ☆432Updated 9 months ago
• Tony-Y / pytorch_warmup
  Learning Rate Warmup in PyTorch
  ☆411Updated 3 weeks ago
• wzlxjtu / PositionalEncoding2D
  A PyTorch implementation of the 1d and 2d Sinusoidal positional encoding/embedding.
  ☆252Updated 4 years ago
• fkodom / fft-conv-pytorch
  Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.
  ☆501Updated last year
• lucidrains / linformer
  Implementation of Linformer for Pytorch
  ☆290Updated last year
• SHI-Labs / Neighborhood-Attention-Transformer
  Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022
  ☆1,131Updated last year
• Sara-Ahmed / SiT
  Self-supervised vIsion Transformer (SiT)
  ☆336Updated 2 years ago
• lucidrains / x-unet
  Implementation of a U-net complete with efficient attention as well as the latest research findings
  ☆282Updated last year
• lucidrains / pixel-level-contrastive-learning
  Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch
  ☆259Updated 4 years ago
• raoyongming / GFNet
  [NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classification
  ☆481Updated 2 years ago
• SHI-Labs / Compact-Transformers
  Escaping the Big Data Paradigm with Compact Transformers, 2021 (Train your Vision Transformers in 30 mins on CIFAR-10 with a single GPU!)
  ☆534Updated 8 months ago
• facebookresearch / convit
  Code for the Convolutional Vision Transformer (ConViT)
  ☆466Updated 3 years ago
• lucidrains / ema-pytorch
  A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
  ☆595Updated 7 months ago
• rishikksh20 / convolution-vision-transformers
  PyTorch Implementation of CvT: Introducing Convolutions to Vision Transformers
  ☆226Updated 4 years ago
• chinhsuanwu / coatnet-pytorch
  A PyTorch implementation of "CoAtNet: Marrying Convolution and Attention for All Data Sizes"
  ☆388Updated 3 years ago
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆784Updated last year
• lucidrains / local-attention
  An implementation of local windowed attention for language modeling
  ☆460Updated 6 months ago
• katsura-jp / pytorch-cosine-annealing-with-warmup
  ☆462Updated 2 years ago
• microsoft / CvT
  This is an official implementation of CvT: Introducing Convolutions to Vision Transformers.
  ☆579Updated 2 years ago
• Gsunshine / Enjoy-Hamburger
  [ICLR 2021 top 3%] Is Attention Better Than Matrix Decomposition?
  ☆333Updated 2 years ago
• xxxnell / how-do-vits-work
  (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"
  ☆818Updated 3 years ago
• SwinTransformer / Transformer-SSL
  This is an official implementation for "Self-Supervised Learning with Swin Transformers".
  ☆657Updated 4 years ago
• lucidrains / segformer-pytorch
  Implementation of Segformer, Attention + MLP neural network for segmentation, in Pytorch
  ☆370Updated 2 years ago
• tatp22 / multidim-positional-encoding
  An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow
  ☆596Updated 8 months ago