microsoft / tensorflow-rematerializationLinks
Implementation of a Tensorflow XLA rematerialization pass
☆15Updated 5 years ago
Alternatives and similar repositories for tensorflow-rematerialization
Users that are interested in tensorflow-rematerialization are comparing it to the libraries listed below
Sorting:
- A library for syntactically rewriting Python programs, pronounced (sinner).☆69Updated 3 years ago
- Distributed ML Optimizer☆32Updated 3 years ago
- DLPack for Tensorflow☆35Updated 5 years ago
- Standalone commandline CLI tool for compiling Triton kernels☆17Updated 9 months ago
- Make triton easier☆46Updated last year
- A basic Docker-based installation of TVM☆11Updated 3 years ago
- ☆12Updated 3 years ago
- 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
- Some microbenchmarks and design docs before commencement☆12Updated 4 years ago
- A place to store reusable transformer components of my own creation or found on the interwebs☆56Updated last week
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- Implementation of Neural Arithmetic Logic Units (https://arxiv.org/pdf/1808.00508.pdf)☆31Updated 6 years ago
- 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
- Stride visualizations☆37Updated 7 years ago
- benchmarking some transformer deployments☆26Updated 2 years ago
- ☆52Updated 10 months ago
- ☆24Updated 6 years ago
- Automatically insert nvtx ranges to PyTorch models☆17Updated 4 years ago
- Development repository for integrating FlexFlow (A distributed deep learning framework that supports flexible parallelization strategies)…☆29Updated 3 years ago
- Runtime Tracing Library for TensorFlow☆43Updated 6 years ago
- Deadline-based hyperparameter tuning on RayTune.☆31Updated 5 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆131Updated 3 years ago
- Einsum optimization using opt_einsum and PyTorch FX graph rewriting☆21Updated 3 years ago
- a high performance system for customized-precision distributed deep learning☆12Updated 4 years ago
- ☆28Updated 5 months ago
- Codebase associated with the PyTorch compiler tutorial☆46Updated 5 years ago
- ☆30Updated 3 years ago
- TORCH_LOGS parser for PT2☆43Updated 3 weeks ago
- A Learnable LSH Framework for Efficient NN Training☆31Updated 3 years ago
- An experimental ahead of time compiler for Relay.☆50Updated 5 years ago