Rockmate: an Efficient, Fast, Automatic and Generic Tool for Re-materialization in PyTorch.
Xunyi Zhao, Thotime Le Hellard, Lionel Eyraud-Dubois, Julia Gusak, Olivier Beaumont
Browse the full ICML paper archive.
Xunyi Zhao, Thotime Le Hellard, Lionel Eyraud-Dubois, Julia Gusak, Olivier Beaumont
Browse the full ICML paper archive.