Efficient and Automatic Rematerialization for Pytorch training
Project description
rkGB
rk Graph Builder
This package is the Graph Builder part of the rockmate
framework. It provides the computational graph of any Pytorch Module, by parsing and simplifying the
output of torch.export.
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
rkgb-2.0.0.tar.gz
(4.3 kB
view details)
File details
Details for the file rkgb-2.0.0.tar.gz.
File metadata
- Download URL: rkgb-2.0.0.tar.gz
- Upload date:
- Size: 4.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.0.1 CPython/3.10.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
21ed11374897edfb088ddf7b484b894492daf586942f52a30380536c7d3b62ab
|
|
| MD5 |
621ccd980780e7e9ca8365b26d699485
|
|
| BLAKE2b-256 |
ef336e697b4ca572d5adf10ea74a5ece3a797b41c370a64ebbf6a1d644529936
|
