rockmate

Efficient and Automatic Rematerialization for Pytorch training

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install rockmate

Releases

Version Released Bullseye
Python 3.9
Bookworm
Python 3.11
Files
2.0.0 2024-12-09    
1.0.1 2023-07-25    

Issues with this package?

Page last updated 2024-12-09 18:17:02 UTC