Checkmate: Breaking the Memory Wall with Optimal Tensor Rematerialization

Paras Jain

Published On: March 2, 2020

Presented At/In: n Proceedings of 3rd Conference Machine Learning and Systems 2020 (MLSys 2020)

Download Paper: https://rise.cs.berkeley.edu/wp-content/uploads/2020/04/1910.02653.pdf

Link: https://proceedings.mlsys.org/static/paper_files/mlsys/2020/196-Paper.pdf

Authors: Paras Jain, Ajay Jain, Aniruddha Nrusimha, Amir Gholaminejad, Pieter Abbeel, Kurt Keutzer, Ion Stoica, Joseph Gonzalez