GitHub repository: sarvex/dtr-prototype
Primary language: Python
Description: Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616