sail-sg / Adan
Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
β774Updated 7 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,088Updated 9 months ago
- Implementation of ConvMixer for "Patches Are All You Need? π€·"β1,064Updated 2 years ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch modelβ559Updated 2 months ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence lengthβ738Updated 9 months ago
- PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)β1,312Updated 8 months ago
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"β809Updated 2 years ago
- A method to increase the speed and lower the memory footprint of existing vision transformers.β1,010Updated 8 months ago
- Code release for ConvNeXt V2 modelβ1,614Updated 6 months ago
- PyTorch implementation of MoCo v3 https//arxiv.org/abs/2104.02057β1,242Updated 3 years ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorchβ1,120Updated last year
- Neighborhood Attention Extension. Bringing attention to a neighborhood near you!β406Updated 3 weeks ago
- β604Updated last month
- MetaFormer Baselines for Vision (TPAMI 2024)β443Updated 8 months ago
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(nΒ²) Memory"β370Updated last year
- [ECCV 2022] Official repository for "MaxViT: Multi-Axis Vision Transformer". SOTA foundation models for classification, detection, segmenβ¦β457Updated last year
- An All-MLP solution for Vision, from Google AIβ1,013Updated 5 months ago
- Code for NeurIPS 2022 Paper, "Poisson Flow Generative Models" (PFGM)β859Updated last year
- [ICML 2023] Official PyTorch implementation of Global Context Vision Transformersβ429Updated last year
- This is a collection of our NAS and Vision Transformer work.β1,718Updated 6 months ago
- An implementation of the efficient attention module.β301Updated 4 years ago
- Implementation of Linformer for Pytorchβ266Updated last year
- ConvMAE: Masked Convolution Meets Masked Autoencodersβ491Updated last year
- π¦ Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorchβ2,094Updated 2 months ago
- A PyTorch implementation of MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesisβ551Updated last year
- EfficientFormerV2 [ICCV 2023] & EfficientFormer [NeurIPs 2022]β1,015Updated last year
- This is an official implementation for "SimMIM: A Simple Framework for Masked Image Modeling".β950Updated 2 years ago
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorchβ635Updated 2 months ago
- Official implementation of Cold-Diffusion for different transformations in pytorch.β1,052Updated 2 years ago
- A collection of literature after or concurrent with Masked Autoencoder (MAE) (Kaiming He el al.).β805Updated 7 months ago
- [ICLR 2024] Official PyTorch implementation of FasterViT: Fast Vision Transformers with Hierarchical Attentionβ822Updated 8 months ago