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
• cerebroai / reformers
  Efficient Transformers for research, PyTorch and Tensorflow using Locality Sensitive Hashing
  ☆95Updated 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
• hendrycks / GELUs
  A smoother activation function (undergrad code)
  ☆112Updated 5 years ago
• microsoft / infinibatch
  Efficient, check-pointed data loading for deep learning with massive data sets.
  ☆209Updated 2 years ago
• tnq177 / transformers_without_tears
  Transformers without Tears: Improving the Normalization of Self-Attention
  ☆133Updated last year
• cybertronai / pytorch-lamb
  Implementation of https://arxiv.org/abs/1904.00962
  ☆376Updated 4 years ago
• layer6ai-labs / T-Fixup
  Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"
  ☆89Updated 4 years ago
• facebookresearch / FBTT-Embedding
  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
• gonglinyuan / StackingBERT
  Source code for "Efficient Training of BERT by Progressively Stacking"
  ☆113Updated 6 years ago
• huggingface / pytorch_block_sparse
  Fast Block Sparse Matrices for Pytorch
  ☆549Updated 4 years ago
• stefbraun / rnn_benchmarks
  RNN benchmarks of pytorch, tensorflow and theano
  ☆89Updated 7 years ago
• NVIDIA / framework-reproducibility
  Providing reproducibility in deep learning frameworks
  ☆427Updated last year
• lonePatient / lookahead_pytorch
  pytorch implement of Lookahead Optimizer
  ☆194Updated 3 years ago
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆117Updated last year
• ptillet / torch-blocksparse
  Block-sparse primitives for PyTorch
  ☆160Updated 4 years ago
• OverLordGoldDragon / keras-adamw
  Keras/TF implementation of AdamW, SGDW, NadamW, Warm Restarts, and Learning Rate multipliers
  ☆169Updated 3 years ago
• juliuskunze / jaxnet
  Concise deep learning for JAX
  ☆183Updated 4 years ago
• kaiyuyue / torchshard
  Slicing a PyTorch Tensor Into Parallel Shards
  ☆300Updated 2 months ago
• PetrochukM / HParams
  Configure Python functions explicitly and safely
  ☆127Updated 9 months ago
• laiguokun / Funnel-Transformer
  ☆219Updated 5 years ago
• li-js / gpu_memory_profiling
  Profile the GPU memory usage of every line in a Pytorch code
  ☆83Updated 7 years ago
• tbachlechner / ReZero-examples
  PyTorch Examples repo for "ReZero is All You Need: Fast Convergence at Large Depth"
  ☆62Updated last year
• rosinality / adaptive-softmax-pytorch
  Adaptive Softmax implementation for PyTorch
  ☆81Updated 6 years ago
• ofirpress / PartialShuffle
  ☆14Updated 6 years ago
• robeld / ERNIE
  Open Source Neural Machine Translation in PyTorch
  ☆17Updated 6 years ago
• microsoft / DualLearning
  A dual learning toolkit developed by Microsoft Research
  ☆71Updated 2 years ago
• clovaai / length-adaptive-transformer
  Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)
  ☆102Updated 4 years ago
• TezRomacH / layer-to-layer-pytorch
  PyTorch implementation of L2L execution algorithm
  ☆108Updated 2 years ago