Alternatives and similar repositories for tf2-gradient-checkpointing:

Users that are interested in tf2-gradient-checkpointing are comparing it to the libraries listed below

• ymcui / LAMB_Optimizer_TF
  LAMB Optimizer for Large Batch Training (TensorFlow version)
  ☆120Updated 5 years ago
• titu1994 / keras-LAMB-Optimizer
  Implementation of the LAMB optimizer for Keras from the paper "Reducing BERT Pre-Training Time from 3 Days to 76 Minutes"
  ☆76Updated 5 years ago
• gonglinyuan / StackingBERT
  Source code for "Efficient Training of BERT by Progressively Stacking"
  ☆112Updated 5 years ago
• hendrycks / GELUs
  A smoother activation function (undergrad code)
  ☆108Updated 4 years ago
• stefbraun / rnn_benchmarks
  RNN benchmarks of pytorch, tensorflow and theano
  ☆89Updated 6 years ago
• csrhddlam / pytorch-checkpoint
  ☆165Updated 6 years ago
• tbachlechner / ReZero-examples
  PyTorch Examples repo for "ReZero is All You Need: Fast Convergence at Large Depth"
  ☆62Updated 6 months ago
• microsoft / infinibatch
  Efficient, check-pointed data loading for deep learning with massive data sets.
  ☆205Updated last year
• cerebroai / reformers
  Efficient Transformers for research, PyTorch and Tensorflow using Locality Sensitive Hashing
  ☆93Updated 5 years ago
• rosinality / adaptive-softmax-pytorch
  Adaptive Softmax implementation for PyTorch
  ☆79Updated 5 years ago
• cybertronai / transformer-xl
  Training Transformer-XL on 128 GPUs
  ☆140Updated 4 years ago
• CyberZHG / keras-adaptive-softmax
  Adaptive embedding and softmax
  ☆17Updated 3 years ago
• pytorch-tpu / examples
  This repository contains example code to build models on TPUs
  ☆30Updated last year
• michaelrzhang / lookahead
  Implementation for the Lookahead Optimizer.
  ☆240Updated 2 years ago
• mhmoodlan / cyclic-learning-rate
  Cyclic learning rate TensorFlow implementation.
  ☆66Updated 5 years ago
• layer6ai-labs / T-Fixup
  Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"
  ☆89Updated 3 years ago
• CyberZHG / keras-transformer-xl
  Transformer-XL with checkpoint loader
  ☆68Updated 3 years ago
• khakhulin / compressed-transformer
  Compression of NMT transformer model with tensor methods
  ☆48Updated 5 years ago
• clovaai / length-adaptive-transformer
  Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)
  ☆101Updated 4 years ago
• zbloss / reformer_lm
  a Pytorch implementation of the Reformer Network (https://openreview.net/pdf?id=rkgNKkHtvB)
  ☆53Updated 2 years ago
• rdspring1 / PyTorch_GBW_LM
  PyTorch Language Model for 1-Billion Word (LM1B / GBW) Dataset
  ☆122Updated 5 years ago
• TezRomacH / layer-to-layer-pytorch
  PyTorch implementation of L2L execution algorithm
  ☆107Updated 2 years ago
• nw2190 / TensorFlow_Examples
  Collection of TensorFlow Examples
  ☆37Updated 6 years ago
• PetrochukM / HParams
  Configure Python functions explicitly and safely
  ☆126Updated 2 months ago
• namisan / exdeep-nmt
  ☆32Updated 3 years ago
• ptillet / torch-blocksparse
  Block-sparse primitives for PyTorch
  ☆154Updated 3 years ago
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆115Updated 5 months ago
• ofirpress / PartialShuffle
  ☆14Updated 5 years ago
• IBM / PoWER-BERT
  Method to improve inference time for BERT. This is an implementation of the paper titled "PoWER-BERT: Accelerating BERT Inference via Pro…
  ☆59Updated last year
• iKernels / transformers-lightning
  A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…
  ☆47Updated last year