Search Results for author: Lionel Eyraud

Found 1 papers, 1 papers with code

Rockmate: an Efficient, Fast, Automatic and Generic Tool for Re-materialization in PyTorch

1 code implementation3 Jul 2023 Xunyi Zhao, Théotime Le Hellard, Lionel Eyraud, Julia Gusak, Olivier Beaumont

We show through experiments on many models that Rockmate is as fast as Rotor and as efficient as Checkmate, and that it allows in many cases to obtain a significantly lower memory consumption for activations (by a factor of 2 to 5) for a rather negligible overhead (of the order of 10% to 20%).

Cannot find the paper you are looking for? You can Submit a new open access paper.