Implementation of https://arxiv.org/abs/1904.00962
☆377Dec 9, 2020Updated 5 years ago
Alternatives and similar repositories for pytorch-lamb
Users that are interested in pytorch-lamb are comparing it to the libraries listed below
Sorting:
- A LARS implementation in PyTorch☆353Feb 21, 2020Updated 6 years ago
- Implementations of ideas from recent papers☆391Dec 22, 2020Updated 5 years ago
- Training Transformer-XL on 128 GPUs☆141Jun 11, 2020Updated 5 years ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆87Jan 22, 2021Updated 5 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,163Mar 22, 2024Updated last year
- Transformer training code for sequential tasks☆609Sep 14, 2021Updated 4 years ago
- On the Variance of the Adaptive Learning Rate and Beyond☆2,549Jul 31, 2021Updated 4 years ago
- Semantic segmentation pipeline using Catalyst.☆20Apr 3, 2020Updated 5 years ago
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,123Apr 20, 2022Updated 3 years ago
- Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase☆1,207Dec 22, 2023Updated 2 years ago
- AdamP: Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights (ICLR 2021)☆415Jan 13, 2021Updated 5 years ago
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆683Feb 21, 2020Updated 6 years ago
- PyTorch extensions for high performance and large scale training.☆3,400Apr 26, 2025Updated 10 months ago
- PyTorch layer-by-layer model profiler☆606May 23, 2021Updated 4 years ago
- This repo parallelizes mAP_evaluation using python's multiprocessing module.☆18Apr 14, 2022Updated 3 years ago
- Understanding the Difficulty of Training Transformers☆332May 31, 2022Updated 3 years ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,627Mar 25, 2022Updated 3 years ago
- ☆221Jun 8, 2020Updated 5 years ago
- Slides from various talks I gave☆18Oct 25, 2018Updated 7 years ago
- Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)☆102Nov 2, 2020Updated 5 years ago
- Histopathologic Cancer Detection model based on Kaggle Challenge https://www.kaggle.com/c/histopathologic-cancer-detection (top 1%)☆11Feb 16, 2021Updated 5 years ago
- A GPipe implementation in PyTorch☆862Jul 25, 2024Updated last year
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,928Updated this week
- Open source implementation of "Spreading Vectors for Similarity Search"☆320Aug 13, 2021Updated 4 years ago
- PyTorch original implementation of Cross-lingual Language Model Pretraining.☆2,926Feb 14, 2023Updated 3 years ago
- Training RNNs as Fast as CNNs (https://arxiv.org/abs/1709.02755)☆2,112Jan 4, 2022Updated 4 years ago
- Profiling and inspecting memory in pytorch☆1,078Sep 5, 2025Updated 6 months ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆331Jun 13, 2022Updated 3 years ago
- LAMB Optimizer for Large Batch Training (TensorFlow version)☆121Jan 17, 2020Updated 6 years ago
- Configure Python functions explicitly and safely☆129Nov 18, 2024Updated last year
- An optimizer that trains as fast as Adam and as good as SGD.☆2,910Jul 23, 2023Updated 2 years ago
- pytorch implement of Lookahead Optimizer☆195Jun 20, 2022Updated 3 years ago
- A GPU performance profiling tool for PyTorch models☆510Jul 13, 2021Updated 4 years ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,612Jun 16, 2021Updated 4 years ago
- Stochastic Weight Averaging in PyTorch☆977Aug 1, 2021Updated 4 years ago
- Implementation of the LAMB optimizer for Keras from the paper "Reducing BERT Pre-Training Time from 3 Days to 76 Minutes"☆75Apr 5, 2019Updated 6 years ago
- PyProf2: PyTorch Profiling tool☆82Jun 25, 2020Updated 5 years ago
- Over9000 optimizer☆424Nov 22, 2022Updated 3 years ago
- PyTorch DataLoader processed in multiple remote computation machines for heavy data processings☆68Oct 26, 2019Updated 6 years ago