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"
  ☆75Updated 6 years ago
• cybertronai / transformer-xl
  Training Transformer-XL on 128 GPUs
  ☆140Updated 4 years ago
• gonglinyuan / StackingBERT
  Source code for "Efficient Training of BERT by Progressively Stacking"
  ☆112Updated 5 years ago
• pytorch-tpu / examples
  This repository contains example code to build models on TPUs
  ☆30Updated 2 years ago
• microsoft / infinibatch
  Efficient, check-pointed data loading for deep learning with massive data sets.
  ☆208Updated last year
• lonePatient / lookahead_pytorch
  pytorch implement of Lookahead Optimizer
  ☆190Updated 2 years ago
• rosinality / adaptive-softmax-pytorch
  Adaptive Softmax implementation for PyTorch
  ☆80Updated 6 years ago
• juliuskunze / jaxnet
  Concise deep learning for JAX
  ☆183Updated 4 years ago
• CyberZHG / keras-lookahead
  Lookahead mechanism for optimizers in Keras.
  ☆50Updated 3 years ago
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆117Updated 10 months ago
• csrhddlam / pytorch-checkpoint
  ☆165Updated 6 years ago
• tacchinotacchi / distil-bilstm
  Scripts to train a bidirectional LSTM with knowledge distillation from BERT
  ☆158Updated 5 years ago
• clovaai / length-adaptive-transformer
  Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)
  ☆101Updated 4 years ago
• kklemon / keras-loves-torchtext
  Make Torchtext work with Keras.
  ☆18Updated 6 years ago
• allenai / tpu_pretrain
  LM Pretraining with PyTorch/TPU
  ☆134Updated 5 years ago
• bojone / accum_optimizer_for_keras
  wrapping a keras optimizer to implement gradient accumulation
  ☆119Updated 4 years ago
• rdspring1 / PyTorch_GBW_LM
  PyTorch Language Model for 1-Billion Word (LM1B / GBW) Dataset
  ☆123Updated 5 years ago
• tnq177 / transformers_without_tears
  Transformers without Tears: Improving the Normalization of Self-Attention
  ☆131Updated last year
• GLambard / AdamW_Keras
  AdamW optimizer for Keras
  ☆115Updated 5 years ago
• zbloss / reformer_lm
  a Pytorch implementation of the Reformer Network (https://openreview.net/pdf?id=rkgNKkHtvB)
  ☆53Updated 2 years ago
• stefbraun / rnn_benchmarks
  RNN benchmarks of pytorch, tensorflow and theano
  ☆89Updated 6 years ago
• OverLordGoldDragon / keras-adamw
  Keras/TF implementation of AdamW, SGDW, NadamW, Warm Restarts, and Learning Rate multipliers
  ☆167Updated 3 years ago
• allenai / sledgehammer
  ☆47Updated 4 years ago
• PetrochukM / HParams
  Configure Python functions explicitly and safely
  ☆126Updated 6 months ago
• cerebroai / reformers
  Efficient Transformers for research, PyTorch and Tensorflow using Locality Sensitive Hashing
  ☆95Updated 5 years ago
• zphang / bert_on_stilts
  Fork of huggingface/pytorch-pretrained-BERT for BERT on STILTs
  ☆107Updated 2 years ago
• tbachlechner / ReZero-examples
  PyTorch Examples repo for "ReZero is All You Need: Fast Convergence at Large Depth"
  ☆62Updated 10 months ago
• ofirpress / sandwich_transformer
  This repository contains the code for running the character-level Sandwich Transformers from our ACL 2020 paper on Improving Transformer …
  ☆55Updated 4 years ago
• mcarilli / mixed_precision_references
  Personal collection of references for high performance mixed precision training.
  ☆41Updated 5 years ago