majumderb / rezero
Official PyTorch Repo for "ReZero is All You Need: Fast Convergence at Large Depth"
☆408Updated 9 months ago
Alternatives and similar repositories for rezero:
Users that are interested in rezero are comparing it to the libraries listed below
- lookahead optimizer (Lookahead Optimizer: k steps forward, 1 step back) for pytorch☆335Updated 5 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆374Updated 4 years ago
- pytorch implement of Lookahead Optimizer☆189Updated 2 years ago
- DeLighT: Very Deep and Light-Weight Transformers☆467Updated 4 years ago
- Accelerate training by storing parameters in one contiguous chunk of memory.☆291Updated 4 years ago
- [ICLR 2020] Lite Transformer with Long-Short Range Attention☆607Updated 9 months ago
- My take on a practical implementation of Linformer for Pytorch.☆414Updated 2 years ago
- Source code for "On the Relationship between Self-Attention and Convolutional Layers"☆1,099Updated 2 years ago
- Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention☆262Updated 3 years ago
- Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase☆1,201Updated last year
- A New Optimization Technique for Deep Neural Networks☆535Updated 3 years ago
- Implementation for the Lookahead Optimizer.☆241Updated 3 years ago
- Apollo: An Adaptive Parameter-wise Diagonal Quasi-Newton Method for Nonconvex Stochastic Optimization☆183Updated 3 years ago
- Fully featured implementation of Routing Transformer☆291Updated 3 years ago
- Understanding the Difficulty of Training Transformers☆329Updated 2 years ago
- Sublinear memory optimization for deep learning. https://arxiv.org/abs/1604.06174☆598Updated 5 years ago
- Transformer with Untied Positional Encoding (TUPE). Code of paper "Rethinking Positional Encoding in Language Pre-training". Improve exis…☆251Updated 3 years ago
- Implementations of ideas from recent papers☆393Updated 4 years ago
- [ICML 2021 Oral] We show pure attention suffers rank collapse, and how different mechanisms combat it.☆164Updated 4 years ago
- Debug PyTorch code using PySnooper☆799Updated 4 years ago
- Implements https://arxiv.org/abs/1711.05101 AdamW optimizer, cosine learning rate scheduler and "Cyclical Learning Rates for Training Neu…☆148Updated 5 years ago
- Deep Learning Experiment Management☆640Updated 2 years ago
- Repository for NeurIPS 2020 Spotlight "AdaBelief Optimizer: Adapting stepsizes by the belief in observed gradients"☆1,060Updated 8 months ago
- Over9000 optimizer☆426Updated 2 years ago
- Transformer training code for sequential tasks☆612Updated 3 years ago
- Implementation of LambdaNetworks, a new approach to image recognition that reaches SOTA with less compute☆1,531Updated 4 years ago
- Code for Noisy Student Training. https://arxiv.org/abs/1911.04252☆761Updated 4 years ago
- Code for "Gradient Surgery for Multi-Task Learning"☆320Updated 5 years ago
- torchsummaryX: Improved visualization tool of torchsummary☆303Updated 3 years ago
- PyTorch layer-by-layer model profiler☆607Updated 3 years ago