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

Metadata

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

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.7
File Size Uploaded
orbax_checkpoint-0.12.7.tar.gz 762.9 kB Details

Built distribution (wheel)

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

Total release size: 2.2 MB

Release files / orbax_checkpoint-0.12.7.tar.gz

Download URL orbax_checkpoint-0.12.7.tar.gz
Size 762.9 kB
Tags Source
SHA-256 checksum
How to use checksums 		96b352c3d017bf268a05977ea7677ce5af1faa728a99204ea3dc8157c6ce575a
BLAKE2b-256 checksum
How to use checksums 		09313c9cc5ba3abde389064cc165da0a7f874e4f0402d929a5ff84900d94b8d0
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.7-py3-none-any.whl

Download URL orbax_checkpoint-0.12.7-py3-none-any.whl
Size 1.4 MB
Tags Python 3
SHA-256 checksum
How to use checksums 		ed62a3efcdbce9639a6f6ecbefc050ced035f6fb90e578033deffd492983da92
BLAKE2b-256 checksum
How to use checksums 		f73ffe0843d2cc752e1ec551dddfc598d5193ec7327995bf417f35fcb94c7dc4
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.7 This release

2 release files

0.12.6

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