davisyoshida / tf2-gradient-checkpointingLinks
Simple gradient checkpointing for eager mode execution
☆46Updated 4 years ago
Alternatives and similar repositories for tf2-gradient-checkpointing
Users that are interested in tf2-gradient-checkpointing are comparing it to the libraries listed below
Sorting:
- LAMB Optimizer for Large Batch Training (TensorFlow version)☆120Updated 5 years ago
- Efficient Transformers for research, PyTorch and Tensorflow using Locality Sensitive Hashing☆95Updated 5 years ago
- Training Transformer-XL on 128 GPUs☆140Updated 5 years ago
- Implementation of the LAMB optimizer for Keras from the paper "Reducing BERT Pre-Training Time from 3 Days to 76 Minutes"☆75Updated 6 years ago
- Efficient, check-pointed data loading for deep learning with massive data sets.☆208Updated 2 years ago
- RNN benchmarks of pytorch, tensorflow and theano☆89Updated 7 years ago
- Source code for "Efficient Training of BERT by Progressively Stacking"☆112Updated 6 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆376Updated 4 years ago
- Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"☆89Updated 4 years ago
- ☆14Updated 6 years ago
- ☆218Updated 5 years ago
- Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)☆101Updated 4 years ago
- This repository contains the code for running the character-level Sandwich Transformers from our ACL 2020 paper on Improving Transformer …☆55Updated 4 years ago
- Compression of NMT transformer model with tensor methods☆48Updated 6 years ago
- PyTorch Language Model for 1-Billion Word (LM1B / GBW) Dataset☆123Updated 5 years ago
- Method to improve inference time for BERT. This is an implementation of the paper titled "PoWER-BERT: Accelerating BERT Inference via Pro…☆62Updated 3 months ago
- Source code of paper "BP-Transformer: Modelling Long-Range Context via Binary Partitioning"☆128Updated 4 years ago
- A smoother activation function (undergrad code)☆112Updated 5 years ago
- a Pytorch implementation of the Reformer Network (https://openreview.net/pdf?id=rkgNKkHtvB)☆53Updated 2 years ago
- Scripts to train a bidirectional LSTM with knowledge distillation from BERT☆158Updated 5 years ago
- Transformers without Tears: Improving the Normalization of Self-Attention☆133Updated last year
- This is a Tensor Train based compression library to compress sparse embedding tables used in large-scale machine learning models such as …☆194Updated 3 years ago
- Transformer-XL with checkpoint loader☆68Updated 3 years ago
- Cascaded Text Generation with Markov Transformers☆129Updated 2 years ago
- Understanding the Difficulty of Training Transformers☆329Updated 3 years ago
- PyTorch implementation of L2L execution algorithm☆107Updated 2 years ago
- Torch Distributed Experimental☆117Updated last year
- Block-sparse primitives for PyTorch☆157Updated 4 years ago
- ☆32Updated 3 years ago
- Knowledge Distillation For Transformer Language Models☆52Updated last year