Metadata-Version: 2.1
Name: orbax
Version: 0.0.15
Summary: Orbax
Home-page: http://github.com/google/orbax
Author: Google Inc.
Author-email: no-reply@google.com
License: Apache 2.0
Keywords: JAX machine learning
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Description-Content-Type: text/markdown
Provides-Extra: test
License-File: LICENSE

# Orbax

[Orbax](https://github.com/google/orbax/blob/main/docs/index.md) 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](https://github.com/google/orbax/blob/main/docs/checkpoint.md).

Check out our [colab](http://colab.research.google.com/github/google/orbax/blob/main/orbax//checkpoint/orbax_checkpoint.ipynb) for a hands-on introduction.

