Skip to main content

Set of libraries for ML training loops in JAX.

Project description

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.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

clu-0.0.12.tar.gz (72.7 kB view details)

Uploaded Source

Built Distribution

clu-0.0.12-py3-none-any.whl (101.8 kB view details)

Uploaded Python 3

File details

Details for the file clu-0.0.12.tar.gz.

File metadata

  • Download URL: clu-0.0.12.tar.gz
  • Upload date:
  • Size: 72.7 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.0.0 CPython/3.12.2

File hashes

Hashes for clu-0.0.12.tar.gz
Algorithm Hash digest
SHA256 f71eaa1afbd30f57f7709257ba7e1feb8ad5c1c3dcae3606672a138678bb3ce4
MD5 a5ce28213e6976d83b66440afc4a5425
BLAKE2b-256 77b209a9ab02dc856b2edff2085259c2fe6830b728f5a78e89ffaa2a763002cc

See more details on using hashes here.

File details

Details for the file clu-0.0.12-py3-none-any.whl.

File metadata

  • Download URL: clu-0.0.12-py3-none-any.whl
  • Upload date:
  • Size: 101.8 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.0.0 CPython/3.12.2

File hashes

Hashes for clu-0.0.12-py3-none-any.whl
Algorithm Hash digest
SHA256 0d183e7d25f7dd0700444510a264e24700e2f068bdabd199ed22866f7e54edba
MD5 e13b6cbd36e7a1d02cc276f317264688
BLAKE2b-256 08048651aacf8a55b1cd46f620e629c6c1658181dacb684b5a09aadda32ff11d

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page