A PyTorch library that allows tensor memory to be temporarily released and resumed later.
conda install conda-forge::torch_memory_saver