Alternatives and similar repositories for awesome-mlp-mixer:

Users that are interested in awesome-mlp-mixer are comparing it to the libraries listed below

• ziplab / EcoFormer
  [NeurIPS 2022 Spotlight] This is the official PyTorch implementation of "EcoFormer: Energy-Saving Attention with Linear Complexity"
  ☆70Updated 2 years ago
• rjbruin / flexconv
  Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…
  ☆115Updated 2 years ago
• VITA-Group / ViT-Anti-Oversmoothing
  [ICLR 2022] "Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practice" by Peihao Wang, Wen…
  ☆80Updated last year
• haofanwang / awesome-mlp-papers
  Recent Advances in MLP-based Models (MLP is all you need!)
  ☆114Updated 2 years ago
• karttikeya / minREV
  A simple minimal implementation of Reversible Vision Transformers
  ☆122Updated last year
• google-research / diffstride
  TF/Keras code for DiffStride, a pooling layer with learnable strides.
  ☆125Updated 3 years ago
• FocalNet / Networks-Beyond-Attention
  A compilation of network architectures for vision and others without usage of self-attention mechanism
  ☆77Updated 2 years ago
• willGuimont / learnable_fourier_positional_encoding
  Learnable Fourier Features for Multi-Dimensional Spatial Positional Encoding
  ☆45Updated 5 months ago
• SamsungLabs / ASAM
  Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.
  ☆141Updated 3 years ago
• TomerRonen34 / mixed-resolution-vit
  ☆50Updated last year
• IBM / RegionViT
  open source the research work for published on arxiv. https://arxiv.org/abs/2106.02689
  ☆51Updated 3 years ago
• jokofa / torch_kmeans
  PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.
  ☆63Updated 2 years ago
• yingyichen-cyy / PrimalAttention
  (NeurIPS 2023) PyTorch implementation of "Primal-Attention: Self-attention through Asymmetric Kernel SVD in Primal Representation"
  ☆18Updated 5 months ago
• rishikksh20 / ResMLP-pytorch
  ResMLP: Feedforward networks for image classification with data-efficient training
  ☆42Updated 3 years ago
• okojoalg / raft-mlp
  ☆25Updated 3 years ago
• lucidrains / flash-cosine-sim-attention
  Implementation of fused cosine similarity attention in the same style as Flash Attention
  ☆211Updated 2 years ago
• enyac-group / supmae
  This is a offical PyTorch/GPU implementation of SupMAE.
  ☆77Updated 2 years ago
• JeongsooP / RGB-no-more
  An official code release of the paper RGB no more: Minimally Decoded JPEG Vision Transformers
  ☆56Updated last year
• GATECH-EIC / Castling-ViT
  [CVPR 2023] Castling-ViT: Compressing Self-Attention via Switching Towards Linear-Angular Attention During Vision Transformer Inference
  ☆29Updated last year
• NVlabs / ConvSSM
  ☆65Updated 4 months ago
• Felix-Petersen / difftopk
  Differentiable Top-k Classification Learning
  ☆80Updated 2 years ago
• Meituan-AutoML / CPVT
  ☆191Updated 2 years ago
• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆86Updated last year
• lucidrains / bidirectional-cross-attention
  A simple cross attention that updates both the source and target in one step
  ☆164Updated 10 months ago
• ethanbar11 / ssm_2d
  More dimensions = More fun
  ☆21Updated 7 months ago
• rishikksh20 / MLP-Mixer-pytorch
  Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision
  ☆217Updated 3 years ago
• moabarar / qna
  [CVPR2022 - Oral] Official Jax Implementation of Learned Queries for Efficient Local Attention
  ☆116Updated 2 years ago
• sndnyang / Diffusion_ViT
  PyTorch Implementation of "Your ViT is Secretly a Hybrid Discriminative-Generative Diffusion Model"
  ☆48Updated 2 years ago
• facebookresearch / RCDM
  Visualizing representations with diffusion based conditional generative model.
  ☆90Updated last year
• Haochen-Wang409 / DropPos
  [NeurIPS'23] DropPos: Pre-Training Vision Transformers by Reconstructing Dropped Positions
  ☆61Updated 10 months ago