lucidrains / axial-attention

Related projects ⓘ

Alternatives and complementary repositories for axial-attention

• csrhddlam / axial-deeplab
  This is a PyTorch re-implementation of Axial-DeepLab (ECCV 2020 Spotlight)
  ☆450Updated 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…
  ☆300Updated 2 years ago
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆1,003Updated 2 months ago
• lucidrains / pixel-level-contrastive-learning
  Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch
  ☆252Updated 3 years ago
• cmsflash / efficient-attention
  An implementation of the efficient attention module.
  ☆284Updated 3 years ago
• Sara-Ahmed / SiT
  Self-supervised vIsion Transformer (SiT)
  ☆325Updated last year
• lucidrains / linformer
  Implementation of Linformer for Pytorch
  ☆257Updated 10 months ago
• rishikksh20 / MLP-Mixer-pytorch
  Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision
  ☆209Updated 3 years ago
• SwinTransformer / Transformer-SSL
  This is an official implementation for "Self-Supervised Learning with Swin Transformers".
  ☆627Updated 3 years ago
• wzlxjtu / PositionalEncoding2D
  A PyTorch implementation of the 1d and 2d Sinusoidal positional encoding/embedding.
  ☆250Updated 4 years ago
• microsoft / Focal-Transformer
  [NeurIPS 2021 Spotlight] Official code for "Focal Self-attention for Local-Global Interactions in Vision Transformers"
  ☆546Updated 2 years ago
• facebookresearch / convit
  Code for the Convolutional Vision Transformer (ConViT)
  ☆462Updated 3 years ago
• rishikksh20 / convolution-vision-transformers
  PyTorch Implementation of CvT: Introducing Convolutions to Vision Transformers
  ☆224Updated 3 years ago
• microsoft / CvT
  This is an official implementation of CvT: Introducing Convolutions to Vision Transformers.
  ☆555Updated last year
• lucidrains / local-attention
  An implementation of local windowed attention for language modeling
  ☆384Updated 2 months ago
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆698Updated 6 months ago
• berniwal / swin-transformer-pytorch
  Implementation of the Swin Transformer in PyTorch.
  ☆796Updated 3 years ago
• raoyongming / GFNet
  [NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classification
  ☆445Updated last year
• microsoft / esvit
  EsViT: Efficient self-supervised Vision Transformers
  ☆408Updated last year
• fadel / pytorch_ema
  Tiny PyTorch library for maintaining a moving average of a collection of parameters.
  ☆406Updated last month
• locuslab / convmixer
  Implementation of ConvMixer for "Patches Are All You Need? 🤷"
  ☆1,062Updated 2 years ago
• lucidrains / perceiver-pytorch
  Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch
  ☆1,097Updated last year
• lucidrains / x-unet
  Implementation of a U-net complete with efficient attention as well as the latest research findings
  ☆267Updated 6 months ago
• tatp22 / multidim-positional-encoding
  An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow
  ☆546Updated 3 weeks ago
• lucidrains / bidirectional-cross-attention
  A simple cross attention that updates both the source and target in one step
  ☆152Updated 6 months ago
• Meituan-AutoML / Twins
  Two simple and effective designs of vision transformer, which is on par with the Swin transformer
  ☆580Updated last year
• lucidrains / deformable-attention
  Implementation of Deformable Attention in Pytorch from the paper "Vision Transformer with Deformable Attention"
  ☆287Updated 6 months ago
• rishikksh20 / FNet-pytorch
  Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms
  ☆251Updated 3 years ago
• Tony-Y / pytorch_warmup
  Learning Rate Warmup in PyTorch
  ☆392Updated this week
• IBM / CrossViT
  Official implementation of CrossViT. https://arxiv.org/abs/2103.14899
  ☆353Updated 2 years ago