Alternatives and similar repositories for tensorflow-rematerialization:

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

• octoml / synr
  A library for syntactically rewriting Python programs, pronounced (sinner).
  ☆70Updated 2 years ago
• ray-project / distml
  Distributed ML Optimizer
  ☆30Updated 3 years ago
• facebookresearch / DLRM-FlexFlow
  Development repository for integrating FlexFlow (A distributed deep learning framework that supports flexible parallelization strategies)…
  ☆28Updated 3 years ago
• zdevito / custom_loader
  ☆12Updated 3 years ago
• nod-ai / transformer-benchmarks
  benchmarking some transformer deployments
  ☆26Updated last year
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆12Updated 3 years ago
• VoVAllen / tf-dlpack
  DLPack for Tensorflow
  ☆36Updated 4 years ago
• microsoft / DGT
  Learning Accurate Decision Trees with Bandit Feedback via Quantized Gradient Descent
  ☆14Updated 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 …
  ☆63Updated 2 years ago
• microsoft / torchy
  A tracing JIT for PyTorch
  ☆17Updated 2 years ago
• bwasti / pytorch_compiler_tutorial
  Codebase associated with the PyTorch compiler tutorial
  ☆44Updated 5 years ago
• hogepodge / tvm-docker
  A basic Docker-based installation of TVM
  ☆12Updated 2 years ago
• UmerHA / triton_util
  Make triton easier
  ☆42Updated 7 months 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
  ☆36Updated 6 years ago
• drcut / CPD
  a high performance system for customized-precision distributed deep learning
  ☆12Updated 4 years ago
• zhisbug / ray-scalable-ml-design
  Some microbenchmarks and design docs before commencement
  ☆12Updated 3 years ago
• iree-org / iree-jax
  ☆50Updated 5 months ago
• manishucsd / py-codegen
  ☆15Updated 3 months ago
• microsoft / dist-ir
  An IR for efficiently simulating distributed ML computation.
  ☆25Updated last year
• mikex86 / tritonc
  Standalone commandline CLI tool for compiling Triton kernels
  ☆17Updated 4 months ago
• joschu / jax-exp
  ☆24Updated 5 years ago
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated 3 months ago
• HazyResearch / mongoose
  A Learnable LSH Framework for Efficient NN Training
  ☆30Updated 3 years ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆130Updated 2 years ago
• facebookresearch / loop_nest
  Loop Nest - Linear algebra compiler and code generator.
  ☆22Updated 2 years ago
• distdl / distdl
  ☆21Updated 2 years ago
• jxbz / nero
  👑 Pytorch code for the Nero optimiser.
  ☆20Updated 2 years ago
• srush / transformers-bet
  ☆12Updated 2 years ago
• sail-sg / hloenv
  an environment based on XLA for deep learning compiler optimization research.
  ☆23Updated last year