jaketae / mlp-mixerLinks
PyTorch implementation of MLP-Mixer: An all-MLP Architecture for Vision
☆29Updated 4 years ago
Alternatives and similar repositories for mlp-mixer
Users that are interested in mlp-mixer are comparing it to the libraries listed below
Sorting:
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆142Updated 4 years ago
- DeltaCNN End-to-End CNN Inference of Sparse Frame Differences in Videos☆59Updated 2 years ago
- Implementation of Uniformer, a simple attention and 3d convolutional net that achieved SOTA in a number of video classification tasks, de…☆102Updated 3 years ago
- This is the official PyTorch implementation for "Mesa: A Memory-saving Training Framework for Transformers".☆121Updated 4 years ago
- Implementation of Kronecker Attention in Pytorch☆19Updated 5 years ago
- [NeurIPS 2022 Spotlight] This is the official PyTorch implementation of "EcoFormer: Energy-Saving Attention with Linear Complexity"☆73Updated 3 years ago
- ImageNet-12k subset of ImageNet-21k (fall11)☆21Updated 2 years ago
- A simple minimal implementation of Reversible Vision Transformers☆126Updated last year
- Implementation of OmniNet, Omnidirectional Representations from Transformers, in Pytorch☆59Updated 4 years ago
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆51Updated 3 years ago
- [NeurIPS'21] "Ultra-Data-Efficient GAN Training: Drawing A Lottery Ticket First, Then Training It Toughly", Tianlong Chen, Yu Cheng, Zhe …☆84Updated 4 years ago
- [ICLR 2022] "As-ViT: Auto-scaling Vision Transformers without Training" by Wuyang Chen, Wei Huang, Xianzhi Du, Xiaodan Song, Zhangyang Wa…☆76Updated 3 years ago
- Transformers w/o Attention, based fully on MLPs☆97Updated last year
- [ICLR 2022] "Deep AutoAugment" by Yu Zheng, Zhi Zhang, Shen Yan, Mi Zhang☆65Updated last year
- [ICCV 2021] A Pytorch implementation of "Manifold Matching via Deep Metric Learning for Generative Modeling"☆80Updated 2 years ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆116Updated 3 years ago
- ☆88Updated 2 years ago
- (ECCV 2022) BayesCap: Bayesian Identity Cap for Calibrated Uncertainty in Frozen Neural Networks☆51Updated 3 years ago
- ☆33Updated 3 years ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆99Updated 5 years ago
- An open source implementation of CLIP.☆33Updated 3 years ago
- understanding model mistakes with human annotations☆106Updated 2 years ago
- State-of-the-art data augmentation search algorithms in PyTorch☆47Updated 2 years ago
- The PyTorch implementation of Latent Video Transformer.☆99Updated last year
- ☆42Updated 2 years ago
- Faster depthwise convolutions for PyTorch☆66Updated 5 years ago
- Simple notebooks to learn diffusion models on toy datasets☆17Updated 2 years ago
- Unofficial PyTorch implementation of TokenLearner by Google AI☆66Updated 3 years ago
- [WACV 2022] "Sandwich Batch Normalization: A Drop-In Replacement for Feature Distribution Heterogeneity" by Xinyu Gong, Wuyang Chen, Tian…☆51Updated 4 years ago
- Check gradient flow in Pytorch☆85Updated 7 years ago