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 links

Key dates

PyPI data
Data sourced directly from PyPI's database.
  • Released:
Newer release available (0.12.6)

1 maintainer

PyPI data
Data sourced directly from PyPI's database.
Avatar for orbax from gravatar.com orbax

Credits

Author: Orbax Authors

License

Apache Software License

Requires

Python >=3.11

Provides Extra

docs testing tiering-service

Tags

JAX machine learning checkpoint training

Classifiers

Report project as malware

Release files for orbax-checkpoint 0.12.5

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for orbax-checkpoint 0.12.5
File Size Uploaded
orbax_checkpoint-0.12.5.tar.gz 757.2 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for orbax-checkpoint 0.12.5
File Interpreter ABI Platform
orbax_checkpoint-0.12.5-py3-none-any.whl 1.4 MB Python 3 none any Details

Total release size: 2.1 MB

Release files / orbax_checkpoint-0.12.5.tar.gz

Download URL orbax_checkpoint-0.12.5.tar.gz
Size 757.2 kB
Tags Source
SHA-256 checksum
How to use checksums 		337b803370f62d23dc85d174fa9fb59115f9f5b43b5473b23318e118fd0c6114
BLAKE2b-256 checksum
How to use checksums 		dde36c49ca1786e55459a6e3738e8425d9604ec4b87fa84b881d9c86454cd941
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing? 		No
Uploaded via twine/7.0.0 CPython/3.12.14

Release files / orbax_checkpoint-0.12.5-py3-none-any.whl

Download URL orbax_checkpoint-0.12.5-py3-none-any.whl
Size 1.4 MB
Tags Python 3
SHA-256 checksum
How to use checksums 		2fa697de73f75c6dd5f94ca84b1a87deefc101a531e60a04bf7574a23c922d03
BLAKE2b-256 checksum
How to use checksums 		3acc7b5b26d16c818919db8c8af9e3d01f731e68b1452447f7f4544abe30ab2f
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing? 		No
Uploaded via twine/7.0.0 CPython/3.12.14

Release history Release notifications | RSS feed

0.12.6

2 release files

This release

0.12.5 This release

2 release files

0.12.4

2 release files

0.12.3

2 release files

0.12.2

2 release files

0.12.1

2 release files

0.12.0

2 release files

0.11.40

2 release files

0.11.39

2 release files

0.11.38

2 release files

0.11.37

2 release files

0.11.36

2 release files

0.11.35

2 release files

0.11.34

2 release files

0.11.33

2 release files

0.11.32

2 release files

0.11.31

2 release files

0.11.30

2 release files

0.11.28

2 release files

0.11.27

2 release files

0.11.26

2 release files

0.11.25

2 release files

0.11.24

2 release files

0.11.23

2 release files

0.11.22

2 release files

0.11.21

2 release files

0.11.20

2 release files

0.11.19

2 release files

0.11.18

2 release files

0.11.17

2 release files

0.11.16

2 release files

0.11.15

2 release files

0.11.14

2 release files

0.11.13

2 release files

0.11.12

2 release files

0.11.11

2 release files

0.11.10

2 release files

0.11.9

2 release files

0.11.8

2 release files

0.11.7

2 release files

0.11.6

2 release files

0.11.5

2 release files

0.11.4

2 release files

0.11.3

2 release files

0.11.2

2 release files

0.11.1

2 release files

0.11.0

2 release files

0.10.3

2 release files

0.10.2

2 release files

0.10.1

2 release files

0.10.0

2 release files

0.9.1

2 release files

0.9.0

2 release files

0.8.0

2 release files

0.7.0

2 release files

0.6.4

2 release files

0.6.3

2 release files

0.6.2

2 release files

0.6.1

2 release files

0.6.0

2 release files

0.5.23

2 release files

0.5.22

2 release files

0.5.21

2 release files

0.5.20

2 release files

0.5.19

2 release files

0.5.18

2 release files

0.5.17

2 release files

0.5.16

2 release files

0.5.15

2 release files

0.5.14

2 release files

0.5.13

2 release files

0.5.12

2 release files

0.5.11

2 release files

0.5.10

2 release files

0.5.9

2 release files

0.5.8

2 release files

0.5.7

2 release files

0.5.6

2 release files

0.5.5

2 release files

0.5.4

2 release files

0.5.3

2 release files

0.5.2

2 release files

0.5.1

2 release files

0.5.0

2 release files

0.4.8

2 release files

0.4.7

2 release files

0.4.6

2 release files

0.4.5

2 release files

0.4.4

2 release files

0.4.3

2 release files

0.4.2

2 release files

0.4.1

2 release files

0.4.0

2 release files

0.3.5

2 release files

0.3.4

2 release files

0.3.3

2 release files

0.3.2

2 release files

0.3.1

2 release files

0.3.0

2 release files

0.2.7

2 release files

0.2.6

2 release files

0.2.5

2 release files

0.2.4

2 release files

0.2.3

2 release files

0.2.2

2 release files

0.2.1

2 release files

0.2.0

2 release files

0.1.8

2 release files

0.1.7

2 release files

0.1.6

2 release files

0.1.4

2 release files

0.1.1

2 release files

0.0.0

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page