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 5 years ago
• TezRomacH / layer-to-layer-pytorch
  PyTorch implementation of L2L execution algorithm
  ☆107Updated 2 years ago
• cerebroai / reformers
  Efficient Transformers for research, PyTorch and Tensorflow using Locality Sensitive Hashing
  ☆95Updated 5 years ago
• stefbraun / rnn_benchmarks
  RNN benchmarks of pytorch, tensorflow and theano
  ☆89Updated 6 years ago
• allenai / tpu_pretrain
  LM Pretraining with PyTorch/TPU
  ☆134Updated 5 years ago
• gonglinyuan / StackingBERT
  Source code for "Efficient Training of BERT by Progressively Stacking"
  ☆112Updated 5 years ago
• OverLordGoldDragon / keras-adamw
  Keras/TF implementation of AdamW, SGDW, NadamW, Warm Restarts, and Learning Rate multipliers
  ☆167Updated 3 years ago
• rdspring1 / PyTorch_GBW_LM
  PyTorch Language Model for 1-Billion Word (LM1B / GBW) Dataset
  ☆123Updated 5 years ago
• lonePatient / lookahead_pytorch
  pytorch implement of Lookahead Optimizer
  ☆190Updated 3 years ago
• ofirpress / PartialShuffle
  ☆14Updated 6 years ago
• laiguokun / Funnel-Transformer
  ☆218Updated 5 years ago
• SeanNaren / minGPT
  A minimal PyTorch Lightning OpenAI GPT w DeepSpeed Training!
  ☆111Updated 2 years ago
• CyberZHG / keras-transformer-xl
  Transformer-XL with checkpoint loader
  ☆68Updated 3 years 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
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆116Updated 10 months ago
• layer6ai-labs / T-Fixup
  Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"
  ☆89Updated 4 years ago
• anandsaha / nips.cocob.pytorch
  PyTorch implementation of the NIPS'17 paper Training Deep Networks without Learning Rates Through Coin Betting.
  ☆37Updated 7 years ago
• tunz / tcop-pytorch
  tunz's CUDA pytorch operator (MaskedSoftmax)
  ☆75Updated 6 years ago
• rosinality / adaptive-softmax-pytorch
  Adaptive Softmax implementation for PyTorch
  ☆81Updated 6 years ago
• juliuskunze / jaxnet
  Concise deep learning for JAX
  ☆183Updated 4 years ago
• ptillet / torch-blocksparse
  Block-sparse primitives for PyTorch
  ☆156Updated 4 years ago
• namisan / exdeep-nmt
  ☆32Updated 3 years ago
• allenai / sledgehammer
  ☆47Updated 5 years ago
• NVlabs / AdaBatch
  AdaBatch: Adaptive Batch Sizes for Training Deep Neural Networks
  ☆42Updated 7 years ago
• khakhulin / compressed-transformer
  Compression of NMT transformer model with tensor methods
  ☆48Updated 6 years ago
• csrhddlam / pytorch-checkpoint
  ☆165Updated 6 years ago
• LiyuanLucasLiu / Transformer-Clinic
  Understanding the Difficulty of Training Transformers
  ☆329Updated 3 years ago
• noahgolmant / pytorch-lars
  "Layer-wise Adaptive Rate Scaling" in PyTorch
  ☆87Updated 4 years ago