Orbax Checkpoint
Project description
Orbax Checkpointing
pip install orbax-checkpoint
(latest PyPi release) OR
pip install 'git+https://github.com/google/orbax/#subdirectory=checkpoint'
(from this repository, at HEAD)
import orbax.checkpoint
Orbax includes a checkpointing library oriented towards JAX users, supporting a variety of different features required by different frameworks, including asynchronous checkpointing, various types, and various storage formats. We aim to provide a highly customizable and composable API which maximizes flexibility for diverse use cases.
Project details
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
Built Distribution
File details
Details for the file orbax_checkpoint-0.9.1.tar.gz
.
File metadata
- Download URL: orbax_checkpoint-0.9.1.tar.gz
- Upload date:
- Size: 212.2 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.11.10
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | edf76d8fc482a9a0296645522f6d13ee09c499af0ef2f9369899cbfee31c7f88 |
|
MD5 | ca8f40aa2d84954bd68390887298786d |
|
BLAKE2b-256 | 48ce4a3386e9e4bc95ab638c2779e259a231ffc2ae1fc2b67a68f4d6d8a794f2 |
File details
Details for the file orbax_checkpoint-0.9.1-py3-none-any.whl
.
File metadata
- Download URL: orbax_checkpoint-0.9.1-py3-none-any.whl
- Upload date:
- Size: 296.5 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.11.10
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | d33e23e63b7ffcf66de3fb5daa3d11bf24a34e66d133eb4a037788f22155857e |
|
MD5 | 99af54e1384ba4473b55447accc5229d |
|
BLAKE2b-256 | 8a4119de59b4a8581ad364970b2429481838cea8d190f50bc8f672ca329b64e0 |