Metadata-Version: 2.4
Name: clu
Version: 0.0.12
Summary: Set of libraries for ML training loops in JAX.
Home-page: http://github.com/google/CommonLoopUtils
Author: Common Loop Utils Authors
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
License-File: AUTHORS
Requires-Dist: absl-py
Requires-Dist: etils[epath,epy]
Requires-Dist: flax
Requires-Dist: jax
Requires-Dist: jaxlib
Requires-Dist: ml_collections
Requires-Dist: numpy
Requires-Dist: packaging
Requires-Dist: typing_extensions
Requires-Dist: wrapt
Provides-Extra: test
Requires-Dist: pytest; extra == "test"
Requires-Dist: tensorflow; extra == "test"
Requires-Dist: tensorflow_datasets; extra == "test"
Requires-Dist: torch>=2.0.0; extra == "test"
Dynamic: author
Dynamic: author-email
Dynamic: classifier
Dynamic: description
Dynamic: description-content-type
Dynamic: home-page
Dynamic: keywords
Dynamic: license
Dynamic: license-file
Dynamic: provides-extra
Dynamic: requires-dist
Dynamic: summary

# CLU - Common Loop Utils

This repository contains common functionality for writing ML training loops. The
goal is to make trainings loops short and readable (but moving common tasks to
small libraries) without removing the flexibility required for research.

To get started, check out this Colab:

https://colab.research.google.com/github/google/CommonLoopUtils/blob/main/clu_synopsis.ipynb

If you're looking for usage examples, see:

https://github.com/google/flax/tree/main/examples

You can also find answers to common questions about CLU on Flax Github
discussions page:

https://github.com/google/flax/discussions

Note: As this point we are not accepting contributions. Please fork the
repository if you want to extend the libraries for your use case.