Rockmate: an Efficient, Fast, Automatic and Generic Tool for Re-materialization in PyTorch. - Xunyi Zhao; Théotime Le Hellard; Lionel Eyraud-Dubois; Julia Gusak; Olivier Beaumont (ICML, 2023)