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:
Latest release

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.6

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.6
File Size Uploaded
orbax_checkpoint-0.12.6.tar.gz 759.6 kB Details

Built distribution (wheel)

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

Total release size: 2.1 MB

Release files / orbax_checkpoint-0.12.6.tar.gz

Download URL orbax_checkpoint-0.12.6.tar.gz
Size 759.6 kB
Tags Source
SHA-256 checksum
How to use checksums 		cdf6a9d03f6c7758089332d2d2ae19434840932da54c9e37b6ef0b96f890a5c1
BLAKE2b-256 checksum
How to use checksums 		c680238ddc4bfb0e52f5ad9d8d893510f732ca9e9df29d58fde4347be5f4ae1b
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.6-py3-none-any.whl

Download URL orbax_checkpoint-0.12.6-py3-none-any.whl
Size 1.4 MB
Tags Python 3
SHA-256 checksum
How to use checksums 		dc512dded6472c0fe0ccde2806f4f8d3026f9bedbbcbfa1ef364f324b13712f0
BLAKE2b-256 checksum
How to use checksums 		45a209c4dd13641d51a3825c15e8e5486d57322634f975831f8cf0e122793c35
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

This release

0.12.6 This release

2 release files

0.12.5

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