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

Open in new window