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