daiszh / dtr-prototype Goto Github PK
View Code? Open in Web Editor NEWThis project forked from uwsampl/dtr-prototype
Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616
Home Page: https://arxiv.org/abs/2006.09616