Set of libraries for ML training loops in JAX.
Project Links
Meta
Author: Common Loop Utils Authors
Classifiers
Development Status
- 4 - Beta
Intended Audience
- Developers
- Science/Research
License
- OSI Approved :: Apache Software License
Programming Language
- Python
Topic
- Scientific/Engineering :: Artificial Intelligence
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.
0.0.12
Apr 10, 2024
0.0.11
Feb 19, 2024
0.0.10
Oct 02, 2023
0.0.9
Apr 26, 2023
0.0.8
Dec 16, 2022
0.0.7
May 20, 2022
0.0.6
Aug 13, 2021
0.0.5
Aug 12, 2021
0.0.4
Jul 01, 2021
0.0.3
Apr 29, 2021
0.0.2
Mar 05, 2021
0.0.1a3
Dec 17, 2020
0.0.1a2
Oct 17, 2020
0.0.1a1
Oct 08, 2020
Wheel compatibility matrix
Files in release
Extras:
Dependencies:
absl-py
etils[epath]
flax
jax
jaxlib
ml-collections
numpy
packaging
typing-extensions
wrapt