Alternatives and similar repositories for tf2-gradient-checkpointing

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

• cerebroai / reformers
  Efficient Transformers for research, PyTorch and Tensorflow using Locality Sensitive Hashing
  ☆95Updated 5 years ago
• ymcui / LAMB_Optimizer_TF
  LAMB Optimizer for Large Batch Training (TensorFlow version)
  ☆121Updated 5 years ago
• microsoft / infinibatch
  Efficient, check-pointed data loading for deep learning with massive data sets.
  ☆211Updated 2 years ago
• cybertronai / transformer-xl
  Training Transformer-XL on 128 GPUs
  ☆141Updated 5 years ago
• stefbraun / rnn_benchmarks
  RNN benchmarks of pytorch, tensorflow and theano
  ☆89Updated 7 years ago
• layer6ai-labs / T-Fixup
  Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"
  ☆89Updated 4 years ago
• tnq177 / transformers_without_tears
  Transformers without Tears: Improving the Normalization of Self-Attention
  ☆134Updated last year
• TezRomacH / layer-to-layer-pytorch
  PyTorch implementation of L2L execution algorithm
  ☆109Updated 2 years ago
• cybertronai / pytorch-lamb
  Implementation of https://arxiv.org/abs/1904.00962
  ☆377Updated 5 years ago
• yzh119 / BPT
  Source code of paper "BP-Transformer: Modelling Long-Range Context via Binary Partitioning"
  ☆127Updated 4 years ago
• rosinality / adaptive-softmax-pytorch
  Adaptive Softmax implementation for PyTorch
  ☆81Updated 6 years ago
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆117Updated last year
• li-js / gpu_memory_profiling
  Profile the GPU memory usage of every line in a Pytorch code
  ☆83Updated 7 years ago
• gonglinyuan / StackingBERT
  Source code for "Efficient Training of BERT by Progressively Stacking"
  ☆113Updated 6 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…
  ☆62Updated 3 months ago
• pytorch-tpu / examples
  This repository contains example code to build models on TPUs
  ☆30Updated 2 years ago
• lonePatient / lookahead_pytorch
  pytorch implement of Lookahead Optimizer
  ☆195Updated 3 years ago
• zbloss / reformer_lm
  a Pytorch implementation of the Reformer Network (https://openreview.net/pdf?id=rkgNKkHtvB)
  ☆53Updated 3 years ago
• ofirpress / PartialShuffle
  ☆14Updated 6 years ago
• clovaai / length-adaptive-transformer
  Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)
  ☆102Updated 5 years ago
• pmichel31415 / are-16-heads-really-better-than-1
  Code for the paper "Are Sixteen Heads Really Better than One?"
  ☆175Updated 5 years ago
• hendrycks / GELUs
  A smoother activation function (undergrad code)
  ☆115Updated 5 years ago
• LiyuanLucasLiu / Transformer-Clinic
  Understanding the Difficulty of Training Transformers
  ☆332Updated 3 years ago
• NVIDIA / dllogger
  A logging tool for deep learning.
  ☆65Updated 9 months ago
• neubig / minnn-assignment
  An assignment on creating a minimalist neural network toolkit for CS11-747
  ☆64Updated 2 years ago
• allenai / sledgehammer
  ☆48Updated 5 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
• VictorZuanazzi / AdaptBatch
  Basic code for adaptive batch in pytorch
  ☆18Updated 6 years ago
• laiguokun / Funnel-Transformer
  ☆220Updated 5 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 …
  ☆57Updated 5 years ago