Skip to main content

Orbax

Project description

Orbax

Orbax is a library providing common utilities for JAX users.

Checkpointing

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.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

orbax-0.0.23.tar.gz (49.6 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

orbax-0.0.23-py3-none-any.whl (66.6 kB view details)

Uploaded Python 3

File details

Details for the file orbax-0.0.23.tar.gz.

File metadata

  • Download URL: orbax-0.0.23.tar.gz
  • Upload date:
  • Size: 49.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.10.8

File hashes

Hashes for orbax-0.0.23.tar.gz
Algorithm Hash digest
SHA256 2d009e579b38a5a94f04bafc01f6d71b0220fc6664c3622bd951f2874e6c0cf1
MD5 af5270d871a412647d1465fcdd7bfe45
BLAKE2b-256 368ee5a7b9b22243accbf6227566bd4bc5f509f7e4d674691560864c1e164464

See more details on using hashes here.

File details

Details for the file orbax-0.0.23-py3-none-any.whl.

File metadata

  • Download URL: orbax-0.0.23-py3-none-any.whl
  • Upload date:
  • Size: 66.6 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.10.8

File hashes

Hashes for orbax-0.0.23-py3-none-any.whl
Algorithm Hash digest
SHA256 870d31d7e6b2aebb502bd6f90670f3f0e53ca4e4563773d6631629920c884744
MD5 2197f1a1fc69b635cfdd94c00d58a4b4
BLAKE2b-256 5cee829a4e3f0e2c5177364750c02a19a709a3206938eabc5ae43973c09c5cab

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page