uwsampl / dtr-prototype
Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616
☆133Updated last year
Alternatives and similar repositories for dtr-prototype:
Users that are interested in dtr-prototype are comparing it to the libraries listed below
- [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration☆197Updated 2 years ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆119Updated 2 years ago
- An Efficient Pipelined Data Parallel Approach for Training Large Model☆74Updated 4 years ago
- ☆141Updated last month
- DietCode Code Release☆61Updated 2 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆130Updated 3 years ago
- ☆37Updated 2 years ago