lucidrains / mlp-mixer-pytorchLinks
An All-MLP solution for Vision, from Google AI
☆1,039Updated 2 months ago
Alternatives and similar repositories for mlp-mixer-pytorch
Users that are interested in mlp-mixer-pytorch are comparing it to the libraries listed below
Sorting:
- Implementation of ConvMixer for "Patches Are All You Need? 🤷"☆1,077Updated 2 years ago
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"☆821Updated 3 years ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆797Updated last year
- Escaping the Big Data Paradigm with Compact Transformers, 2021 (Train your Vision Transformers in 30 mins on CIFAR-10 with a single GPU!)☆535Updated 10 months ago
- PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)☆1,352Updated last year
- PyTorch implementation of SimSiam https//arxiv.org/abs/2011.10566☆1,213Updated 2 years ago
- Vision Transformer (ViT) in PyTorch☆845Updated 3 years ago
- Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch☆1,839Updated last year
- Learning Rate Warmup in PyTorch☆412Updated 2 months ago
- Code for the Convolutional Vision Transformer (ConViT)☆468Updated 3 years ago
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022☆1,145Updated last year
- Implementation of Axial attention - attending to multi-dimensional data efficiently☆385Updated 4 years ago
- PyTorch implementation of MoCo v3 https//arxiv.org/abs/2104.02057☆1,291Updated 3 years ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,171Updated 2 years ago
- Source code for "On the Relationship between Self-Attention and Convolutional Layers"☆1,110Updated 2 years ago
- Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.☆1,210Updated 4 years ago
- This is an official implementation of CvT: Introducing Convolutions to Vision Transformers.☆582Updated 2 years ago
- Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)☆2,075Updated 3 years ago
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision☆218Updated 4 years ago
- An implementation of Performer, a linear attention-based transformer, in Pytorch☆1,148Updated 3 years ago
- ICCV2021, Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNet☆1,190Updated last year
- A PyTorch implementation of "CoAtNet: Marrying Convolution and Attention for All Data Sizes"☆391Updated 3 years ago
- Pytorch library for fast transformer implementations☆1,730Updated 2 years ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆435Updated 11 months ago
- [NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classification☆486Updated 2 years ago
- ☆465Updated 2 years ago
- Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.☆505Updated last year
- SAM: Sharpness-Aware Minimization (PyTorch)☆1,912Updated last year
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆797Updated 3 months ago
- Implementation of Linformer for Pytorch☆298Updated last year