piwheels

rockmate

Efficient and Automatic Rematerialization for Pytorch training

Installation

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

pip3 install rockmate

Dependencies

• configmypy
• torch
• pulp
• rkgb
• numpy

Releases

Version	Released	Bullseye Python 3.9	Bookworm Python 3.11	Trixie Python 3.13	Files
2.0.0	2024-12-09
rockmate-2.0.0-cp311-cp311-linux_armv6l.whl (108 KB) rockmate-2.0.0-cp311-cp311-linux_armv7l.whl (108 KB) rockmate-2.0.0-cp39-cp39-linux_armv6l.whl (108 KB) rockmate-2.0.0-cp39-cp39-linux_armv7l.whl (108 KB) rockmate-2.0.0-cp313-cp313-linux_armv6l.whl (109 KB) rockmate-2.0.0-cp313-cp313-linux_armv7l.whl (109 KB)
1.0.1	2023-07-25
rockmate-1.0.1-cp39-cp39-linux_armv6l.whl (79 KB) rockmate-1.0.1-cp39-cp39-linux_armv7l.whl (79 KB) rockmate-1.0.1-cp311-cp311-linux_armv6l.whl (78 KB) rockmate-1.0.1-cp311-cp311-linux_armv7l.whl (78 KB) rockmate-1.0.1-cp313-cp313-linux_armv6l.whl (80 KB) rockmate-1.0.1-cp313-cp313-linux_armv7l.whl (80 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue