sail-sg / Adan
Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
β769Updated 6 months ago
Alternatives and similar repositories for Adan:
Users that are interested in Adan are comparing it to the libraries listed below
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022β1,077Updated 8 months ago
- Implementation of ConvMixer for "Patches Are All You Need? π€·"β1,063Updated 2 years ago
- PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)β1,308Updated 7 months ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch modelβ548Updated last month
- A method to increase the speed and lower the memory footprint of existing vision transformers.β995Updated 7 months ago
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"β808Updated 2 years ago
- MetaFormer Baselines for Vision (TPAMI 2024)β437Updated 7 months ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence lengthβ724Updated 8 months ago
- [ECCV 2022] Official repository for "MaxViT: Multi-Axis Vision Transformer". SOTA foundation models for classification, detection, segmenβ¦β454Updated last year
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorchβ1,115Updated last year
- Code release for ConvNeXt V2 modelβ1,584Updated 5 months ago
- Neighborhood Attention Extension. Bringing attention to a neighborhood near you!β392Updated 2 weeks ago
- An All-MLP solution for Vision, from Google AIβ1,007Updated 4 months ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.β418Updated 3 months ago
- This is a collection of our NAS and Vision Transformer work.β1,707Updated 5 months ago
- A collection of literature after or concurrent with Masked Autoencoder (MAE) (Kaiming He el al.).β796Updated 6 months ago
- PyTorch implementation of MoCo v3 https//arxiv.org/abs/2104.02057β1,235Updated 3 years ago
- EfficientFormerV2 [ICCV 2023] & EfficientFormer [NeurIPs 2022]β1,006Updated last year
- [ICML 2023] Official PyTorch implementation of Global Context Vision Transformersβ426Updated last year
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorchβ609Updated last month
- This is an official implementation for "SimMIM: A Simple Framework for Masked Image Modeling".β942Updated 2 years ago
- iBOT : Image BERT Pre-Training with Online Tokenizer (ICLR 2022)β696Updated 2 years ago
- [ICLR 2023 Oral] Image as Set of Pointsβ549Updated 8 months ago
- Escaping the Big Data Paradigm with Compact Transformers, 2021 (Train your Vision Transformers in 30 mins on CIFAR-10 with a single GPU!)β509Updated 2 months ago
- A PyTorch implementation of MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesisβ546Updated last year
- β596Updated this week
- π¦ Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorchβ2,080Updated last month
- Pytorch library for fast transformer implementationsβ1,665Updated last year
- [CVPR 2021] Official PyTorch implementation for Transformer Interpretability Beyond Attention Visualization, a novel method to visualize β¦β1,827Updated 11 months ago
- Hiera: A fast, powerful, and simple hierarchical vision transformer.β940Updated 10 months ago