Alternatives and similar repositories for tensorflow-rematerialization:

Users that are interested in tensorflow-rematerialization are comparing it to the libraries listed below

• zhisbug / ray-scalable-ml-design
  Some microbenchmarks and design docs before commencement
  ☆12Updated 4 years ago
• VoVAllen / tf-dlpack
  DLPack for Tensorflow
  ☆35Updated 5 years ago
• octoml / synr
  A library for syntactically rewriting Python programs, pronounced (sinner).
  ☆69Updated 3 years ago
• HazyResearch / mongoose
  A Learnable LSH Framework for Efficient NN Training
  ☆31Updated 3 years ago
• srush / transformers-bet
  ☆12Updated 3 years ago
• jxbz / nero
  👑 Pytorch code for the Nero optimiser.
  ☆20Updated 2 years ago
• ray-project / distml
  Distributed ML Optimizer
  ☆31Updated 3 years ago
• UmerHA / triton_util
  Make triton easier
  ☆47Updated 10 months 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 …
  ☆65Updated 3 years ago
• nod-ai / transformer-benchmarks
  benchmarking some transformer deployments
  ☆26Updated 2 years ago
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆13Updated 3 years ago
• ray-project / ray_shuffling_data_loader
  A Ray-based data loader with per-epoch shuffling and configurable pipelining, for shuffling and loading training data for distributed tra…
  ☆18Updated 2 years ago
• ezyang / stride-visualizer
  Stride visualizations
  ☆37Updated 7 years ago
• hogepodge / tvm-docker
  A basic Docker-based installation of TVM
  ☆11Updated 2 years ago
• facebookresearch / DLRM-FlexFlow
  Development repository for integrating FlexFlow (A distributed deep learning framework that supports flexible parallelization strategies)…
  ☆28Updated 3 years ago
• mikex86 / tritonc
  Standalone commandline CLI tool for compiling Triton kernels
  ☆17Updated 7 months ago
• srush / tangent
  Source-to-Source Debuggable Derivatives in Pure Python
  ☆15Updated last year
• Jokeren / triton-samples
  ☆27Updated 3 months ago
• joschu / jax-exp
  ☆24Updated 6 years ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆130Updated 3 years ago
• drcut / CPD
  a high performance system for customized-precision distributed deep learning
  ☆12Updated 4 years ago
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆19Updated last year
• dglai / minigun
  Light-weight GPU kernel interface for graph operations
  ☆15Updated 4 years ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆78Updated last year
• exists-forall / striped_attention
  ☆38Updated last year
• ahennequ / pytorch-custom-mma
  ☆29Updated 2 years ago
• lianakoleva / no-libtorch-compile
  ☆21Updated last month
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆49Updated this week
• YuhanLiu11 / AutoFreeze
  ☆22Updated 4 years ago
• zdevito / custom_loader
  ☆13Updated 3 years ago