{"categories":[{"label":"Artificial Intelligence","url":"https://skillfed.io/packages/category/scientific-engineering-artificial-intelligence/5"}],"enrichment":{"capability":"CLU provides utilities and abstractions for writing machine learning training loops in JAX, including metrics collection, checkpointing, and common patterns to keep loop code concise while retaining research flexibility.","skillfed_tags":["jax","training-loops","ml-utilities"],"use_cases":["Collecting and aggregating metrics (loss, accuracy, etc.) during JAX training without writing custom metric logic.","Managing checkpoints and model state during long-running training jobs in JAX.","Structuring a custom training loop in Flax or raw JAX while reusing common patterns from CLU.","Prototyping ML experiments in JAX with minimal boilerplate for loop orchestration and logging."],"what_it_does":"CLU is a set of libraries built on top of JAX for structuring machine learning training loops. It abstracts common tasks\u2014metrics aggregation, checkpointing, and loop orchestration\u2014into reusable components so that training code stays readable and concise without sacrificing the flexibility needed for research. The package depends on JAX, Flax, ml-collections, and several utility libraries (absl-py, etils, numpy, wrapt, packaging, typing-extensions) to provide a cohesive toolkit.\n\nThe library is designed for researchers and practitioners writing custom training loops in JAX who want to avoid boilerplate while keeping full control over loop logic. It is actively maintained (last commit 2026-07-07) and widely adopted in the JAX and machine learning community. The project is not currently accepting external contributions but can be forked for custom extensions.","worth_installing":"Yes, if you are writing training loops in JAX or Flax. CLU is actively maintained, has no security vulnerabilities, low install friction, and is widely used in the JAX ecosystem. The permissive Apache 2.0 license poses no restrictions. Verify that CLU's feature set matches your specific checkpoint and metrics needs before committing."},"id":"clu","links":{"html":"https://skillfed.io/packages/clu","md":"https://skillfed.io/packages/clu.md","pypi":"https://pypi.org/project/clu/"},"maintenance":{"status":"active"},"meta":{"latest_release":"2024-04-10","license_spdx":null,"license_treatment":"permissive","name":"clu","python_support":"unspecified","summary":"Set of libraries for ML training loops in JAX."},"popularity":{"monthly_downloads":573809,"position":5944,"tier":"top_15000"},"security":{"n_vulnerabilities":0},"version":"0.0.12"}