sail-sg / Adan
Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
β787Updated 9 months ago
Alternatives and similar repositories for Adan:
Users that are interested in Adan are comparing it to the libraries listed below
- Implementation of ConvMixer for "Patches Are All You Need? π€·"β1,070Updated 2 years ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch modelβ569Updated 4 months ago
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022β1,108Updated 11 months ago
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"β813Updated 2 years ago
- A method to increase the speed and lower the memory footprint of existing vision transformers.β1,042Updated 10 months ago
- PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)β1,324Updated 10 months ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence lengthβ755Updated 11 months ago
- This is a collection of our NAS and Vision Transformer work.β1,740Updated 8 months ago
- iBOT : Image BERT Pre-Training with Online Tokenizer (ICLR 2022)β722Updated 3 years ago
- PyTorch implementation of MoCo v3 https//arxiv.org/abs/2104.02057β1,258Updated 3 years ago
- An All-MLP solution for Vision, from Google AIβ1,017Updated 7 months ago
- This is an official implementation for "SimMIM: A Simple Framework for Masked Image Modeling".β970Updated 2 years ago
- MetaFormer Baselines for Vision (TPAMI 2024)β457Updated 10 months ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.β428Updated 6 months ago
- SAM: Sharpness-Aware Minimization (PyTorch)β1,860Updated last year
- Official PyTorch implementation of Fully Attentional Networksβ475Updated 2 years ago
- Code release for ConvNeXt V2 modelβ1,696Updated 8 months ago
- π¦ Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorchβ2,122Updated 4 months ago
- β621Updated this week
- Escaping the Big Data Paradigm with Compact Transformers, 2021 (Train your Vision Transformers in 30 mins on CIFAR-10 with a single GPU!)β521Updated 5 months ago
- Scaling Up Your Kernels to 31x31: Revisiting Large Kernel Design in CNNs (CVPR 2022)β905Updated 11 months ago
- Code for NeurIPS 2022 Paper, "Poisson Flow Generative Models" (PFGM)β861Updated last year
- Ranger deep learning optimizer rewrite to use newest componentsβ329Updated last year
- Pytorch library for fast transformer implementationsβ1,697Updated 2 years ago
- Pix2Seq codebase: multi-tasks with generative modeling (autoregressive and diffusion)β905Updated last year
- [ECCV 2022] Official repository for "MaxViT: Multi-Axis Vision Transformer". SOTA foundation models for classification, detection, segmenβ¦β470Updated last year
- EfficientFormerV2 [ICCV 2023] & EfficientFormer [NeurIPs 2022]β1,036Updated last year
- ConvMAE: Masked Convolution Meets Masked Autoencodersβ502Updated 2 years ago
- [NeurIPS 2021] [T-PAMI] DynamicViT: Efficient Vision Transformers with Dynamic Token Sparsificationβ600Updated last year
- D-Adaptation for SGD, Adam and AdaGradβ520Updated 2 months ago