Alternatives and similar repositories for tensorflow-rematerialization:

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

• ray-project / distml
  Distributed ML Optimizer
  ☆30Updated 3 years ago
• nod-ai / transformer-benchmarks
  benchmarking some transformer deployments
  ☆26Updated last year
• octoml / synr
  A library for syntactically rewriting Python programs, pronounced (sinner).
  ☆70Updated 3 years ago
• VoVAllen / tf-dlpack
  DLPack for Tensorflow
  ☆36Updated 4 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 …
  ☆63Updated 2 years ago
• UmerHA / triton_util
  Make triton easier
  ☆47Updated 9 months ago
• facebookresearch / DLRM-FlexFlow
  Development repository for integrating FlexFlow (A distributed deep learning framework that supports flexible parallelization strategies)…
  ☆28Updated 3 years ago
• zhisbug / ray-scalable-ml-design
  Some microbenchmarks and design docs before commencement
  ☆12Updated 4 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
• zdevito / custom_loader
  ☆12Updated 3 years ago
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆13Updated 3 years ago
• drcut / CPD
  a high performance system for customized-precision distributed deep learning
  ☆12Updated 4 years ago
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆26Updated 5 months ago
• xldrx / tensorflow-tracer
  Runtime Tracing Library for TensorFlow
  ☆43Updated 6 years ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆130Updated 3 years ago
• exists-forall / striped_attention
  ☆38Updated last year
• mlcommons / training_results_v0.7
  This repository contains the results and code for the MLPerf™ Training v0.7 benchmark.
  ☆56Updated last year
• cchan / tccl
  extensible collectives library in triton
  ☆83Updated 5 months ago
• lianakoleva / no-libtorch-compile
  ☆21Updated 2 weeks ago
• nikitakit / sabertooth
  Standalone pre-training recipe with JAX+Flax
  ☆31Updated last year
• nod-ai / SRT
  Nod.ai 🦈 version of 👻 . You probably want to start at https://github.com/nod-ai/shark for the product and the upstream IREE repository …
  ☆106Updated 2 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆76Updated last year
• jxbz / madam
  👩 Pytorch and Jax code for the Madam optimiser.
  ☆51Updated 4 years ago
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆46Updated 3 months ago
• uwsampl / relay-aot
  An experimental ahead of time compiler for Relay.
  ☆50Updated 4 years ago
• distdl / distdl
  ☆21Updated 2 years ago
• iree-org / iree-jax
  ☆51Updated 7 months ago
• srush / learns-dex
  ☆33Updated 4 years ago
• awslabs / lorien
  ☆44Updated last year
• bwasti / pytorch_compiler_tutorial
  Codebase associated with the PyTorch compiler tutorial
  ☆46Updated 5 years ago