Metadata-Version: 2.1
Name: orbax
Version: 0.0.6
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
License-File: LICENSE
Requires-Dist: absl-py
Requires-Dist: cached-property
Requires-Dist: dataclasses
Requires-Dist: flax
Requires-Dist: jax
Requires-Dist: jaxlib
Requires-Dist: numpy
Requires-Dist: pyyaml
Requires-Dist: tensorflow
Requires-Dist: tensorstore (>=0.1.20)
Provides-Extra: test
Requires-Dist: pytest ; extra == 'test'

Orbax is a library for training utilities for JAX users.
