majumderb / rezero
Official PyTorch Repo for "ReZero is All You Need: Fast Convergence at Large Depth"
☆408Updated 6 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☆334Updated 5 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆371Updated 4 years ago
- Accelerate training by storing parameters in one contiguous chunk of memory.☆292Updated 4 years ago
- pytorch implement of Lookahead Optimizer☆189Updated 2 years ago
- [ICLR 2020] Lite Transformer with Long-Short Range Attention☆606Updated 7 months ago
- Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention☆257Updated 3 years ago
- Apollo: An Adaptive Parameter-wise Diagonal Quasi-Newton Method for Nonconvex Stochastic Optimization☆182Updated 3 years ago
- DeLighT: Very Deep and Light-Weight Transformers☆467Updated 4 years ago
- A New Optimization Technique for Deep Neural Networks☆535Updated 3 years ago
- Transformer training code for sequential tasks☆610Updated 3 years ago
- Source code for "On the Relationship between Self-Attention and Convolutional Layers"☆1,096Updated 2 years ago
- Implementations of ideas from recent papers☆391Updated 4 years ago
- The Noise Contrastive Estimation for softmax output written in Pytorch☆318Updated 5 years ago
- Deep Learning Experiment Management☆639Updated 2 years ago
- This in my Demo of Chen et al. "GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks" ICML 2018☆172Updated 3 years ago
- Transformer with Untied Positional Encoding (TUPE). Code of paper "Rethinking Positional Encoding in Language Pre-training". Improve exis…☆250Updated 3 years ago
- Experimental ground for optimizing memory of pytorch models☆364Updated 6 years ago
- Sublinear memory optimization for deep learning. https://arxiv.org/abs/1604.06174☆594Updated 5 years ago
- Fully featured implementation of Routing Transformer☆288Updated 3 years ago
- Understanding the Difficulty of Training Transformers☆328Updated 2 years ago
- PyTorch layer-by-layer model profiler☆606Updated 3 years ago
- Implements https://arxiv.org/abs/1711.05101 AdamW optimizer, cosine learning rate scheduler and "Cyclical Learning Rates for Training Neu…☆149Updated 5 years ago
- Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase☆1,196Updated last year
- Learning Rate Warmup in PyTorch☆403Updated 2 weeks ago
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆986Updated 4 months ago
- My take on a practical implementation of Linformer for Pytorch.☆411Updated 2 years ago
- Implementation of Sparsemax activation in Pytorch☆158Updated 4 years ago
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆685Updated 4 years ago
- The entmax mapping and its loss, a family of sparse softmax alternatives.☆425Updated 7 months ago
- Stochastic Weight Averaging in PyTorch☆968Updated 3 years ago