Orbax Checkpoint
Project description
Orbax Checkpointing
pip install orbax-checkpoint
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.
To get started, check out our documentation.
Check out our colab for a hands-on introduction.
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
orbax-checkpoint-0.2.0.tar.gz
(60.3 kB
view details)
Built Distribution
File details
Details for the file orbax-checkpoint-0.2.0.tar.gz.
File metadata
- Download URL: orbax-checkpoint-0.2.0.tar.gz
- Upload date:
- Size: 60.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.10.10
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | a4936743b04302ff1b5e1067ba8d6c381fcc6eda4e9f4c2a59bc5301720897d5 |
|
| MD5 | 942f782a73f74c599295e2863101173c |
|
| BLAKE2b-256 | 9571634b28b05864abce014a9fa667e656e2f3a10df1ede85c8c966ec2ed9674 |
Provenance
File details
Details for the file orbax_checkpoint-0.2.0-py3-none-any.whl.
File metadata
- Download URL: orbax_checkpoint-0.2.0-py3-none-any.whl
- Upload date:
- Size: 78.9 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.10.10
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | c56f1fd8527a000aadee2d1ff16caa206f03c9125623edae2a2030b1da58f5ab |
|
| MD5 | 083ad4fbacdf28347caa16fd3eb82dc8 |
|
| BLAKE2b-256 | c88b05b73997edc42a76ab648be0f8bb50d38e7136f14af9b919e8dbf953b5e1 |
