Metadata-Version: 2.1
Name: clu
Version: 0.0.2
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
Platform: UNKNOWN
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
Requires-Dist: absl-py
Requires-Dist: flax
Requires-Dist: jax (>=0.1.76)
Requires-Dist: jaxlib
Requires-Dist: ml-collections
Requires-Dist: numpy (>=1.16.4)
Requires-Dist: tensorflow (>=2.3.0)
Requires-Dist: tensorflow-datasets
Requires-Dist: dataclasses ; python_version < "3.7"
Provides-Extra: pytorch
Requires-Dist: torch (>=1.2.0) ; extra == 'pytorch'
Provides-Extra: test
Requires-Dist: dm-sonnet ; extra == 'test'
Requires-Dist: pytest ; extra == 'test'
Requires-Dist: torch (>=1.2.0) ; extra == 'test'

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

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


