Alternatives and similar repositories for OptimalGradCheckpointing:

Users that are interested in OptimalGradCheckpointing are comparing it to the libraries listed below

• papers-submission / structured_transposable_masks
  Code for ICML 2021 submission
  ☆35Updated 4 years ago
• ucbrise / actnn
  ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training
  ☆200Updated 2 years ago
• cjf00000 / StatQuant
  code for the paper "A Statistical Framework for Low-bitwidth Training of Deep Neural Networks"
  ☆28Updated 4 years ago
• uwsampl / dtr-prototype
  Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616
  ☆132Updated last year
• mlzxy / qsparse
  Train neural networks with joint quantization and pruning on both weights and activations using any pytorch modules
  ☆40Updated 2 years ago
• LiuXiaoxuanPKU / GACT-ICML
  ☆42Updated 2 years ago
• csyhhu / MetaQuant
  Codes for Accepted Paper : "MetaQuant: Learning to Quantize by Learning to Penetrate Non-differentiable Quantization" in NeurIPS 2019
  ☆54Updated 4 years ago
• VITA-Group / UVC
  [ICLR 2022] "Unified Vision Transformer Compression" by Shixing Yu*, Tianlong Chen*, Jiayi Shen, Huan Yuan, Jianchao Tan, Sen Yang, Ji Li…
  ☆52Updated last year
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆62Updated last year
• BradMcDanel / sdgp
  ☆10Updated 3 years ago
• octoml / deformable-attention-kernel
  TVMScript kernel for deformable attention
  ☆25Updated 3 years ago
• yaozhewei / HAP
  ☆43Updated last year
• cli99 / flops-profiler
  pytorch-profiler
  ☆51Updated last year
• moranshkolnik / RobustQuantization
  source code of the paper: Robust Quantization: One Model to Rule Them All
  ☆40Updated 2 years ago
• anony-sub / chameleon
  Chameleon: Adaptive Code Optimization for Expedited Deep Neural Network Compilation
  ☆27Updated 5 years ago
• htqin / BiBERT
  This project is the official implementation of our accepted ICLR 2022 paper BiBERT: Accurate Fully Binarized BERT.
  ☆88Updated last year
• gilshm / sparq
  Post-training sparsity-aware quantization
  ☆34Updated 2 years ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆130Updated 3 years ago
• mit-han-lab / inter-operator-scheduler
  [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration
  ☆196Updated 2 years ago
• aojunzz / NM-sparsity
  ☆226Updated 2 years ago
• facebookresearch / fairring
  Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …
  ☆64Updated 3 years ago
• leeyeehoo / GenNAS
  Generic Neural Architecture Search via Regression (NeurIPS'21 Spotlight)
  ☆36Updated 2 years ago
• GATECH-EIC / CPT
  [ICLR 2021 Spotlight] "CPT: Efficient Deep Neural Network Training via Cyclic Precision" by Yonggan Fu, Han Guo, Meng Li, Xin Yang, Yinin…
  ☆30Updated last year
• megvii-research / SSQL-ECCV2022
  PyTorch implementation of SSQL (Accepted to ECCV2022 oral presentation)
  ☆75Updated 2 years ago
• ModelTC / NNLQP
  ☆35Updated 2 years ago
• Distributed-AI / PipeTransformer
  PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021
  ☆56Updated 3 years ago
• stevenygd / SWALP
  Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".
  ☆62Updated 5 years ago
• XinDongol / DNNAC
  All about acceleration and compression of Deep Neural Networks
  ☆33Updated 5 years ago
• AamirRaihan / SWAT
  Official implementation of Neurips 2020 "Sparse Weight Activation Training" paper.
  ☆27Updated 3 years ago
• snap-research / F8Net
  [ICLR 2022 Oral] F8Net: Fixed-Point 8-bit Only Multiplication for Network Quantization
  ☆94Updated 2 years ago