Alternatives and similar repositories for tensorflow-rematerialization:

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

• 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
• ray-project / distml
  Distributed ML Optimizer
  ☆32Updated 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
• jxbz / nero
  👑 Pytorch code for the Nero optimiser.
  ☆20Updated 2 years ago
• zhisbug / ray-scalable-ml-design
  Some microbenchmarks and design docs before commencement
  ☆12Updated 4 years ago
• drcut / CPD
  a high performance system for customized-precision distributed deep learning
  ☆12Updated 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 …
  ☆65Updated 3 years ago
• srush / transformers-bet
  ☆12Updated 3 years ago
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆13Updated 3 years ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆131Updated 3 years ago
• zdevito / custom_loader
  ☆13Updated 3 years ago
• zasdfgbnm / autonvtx
  Automatically insert nvtx ranges to PyTorch models
  ☆17Updated 4 years ago
• UmerHA / triton_util
  Make triton easier
  ☆47Updated 10 months ago
• bwasti / pytorch_compiler_tutorial
  Codebase associated with the PyTorch compiler tutorial
  ☆45Updated 5 years ago
• xldrx / tensorflow-tracer
  Runtime Tracing Library for TensorFlow
  ☆43Updated 6 years ago
• dglai / minigun
  Light-weight GPU kernel interface for graph operations
  ☆15Updated 4 years ago
• facebookresearch / DLRM-FlexFlow
  Development repository for integrating FlexFlow (A distributed deep learning framework that supports flexible parallelization strategies)…
  ☆28Updated 3 years ago
• merrymercy / NALU
  Implementation of Neural Arithmetic Logic Units (https://arxiv.org/pdf/1808.00508.pdf)
  ☆31Updated 6 years ago
• hogepodge / tvm-docker
  A basic Docker-based installation of TVM
  ☆11Updated 2 years ago
• mikex86 / tritonc
  Standalone commandline CLI tool for compiling Triton kernels
  ☆18Updated 7 months ago
• sail-sg / hloenv
  an environment based on XLA for deep learning compiler optimization research.
  ☆23Updated 2 years ago
• ezyang / stride-visualizer
  Stride visualizations
  ☆37Updated 7 years ago
• uwsampl / relay-aot
  An experimental ahead of time compiler for Relay.
  ☆50Updated 5 years ago
• ucbrise / hypersched
  Deadline-based hyperparameter tuning on RayTune.
  ☆31Updated 5 years ago
• mlcommons / training_results_v0.7
  This repository contains the results and code for the MLPerf™ Training v0.7 benchmark.
  ☆56Updated last year
• joschu / jax-exp
  ☆24Updated 6 years ago
• YuhanLiu11 / AutoFreeze
  ☆22Updated 4 years ago
• feifeibear / PSTensor
  PSTensor provides a way to hack the memory management of tensors in TensorFlow and PyTorch by defining your own C++ Tensor Class.
  ☆10Updated 3 years ago
• percyliang / sfig
  SVG/Javascript-based library for creating presentations/figures
  ☆59Updated 2 years ago