CommonLoopUtils
Machine Learning Model TrainingML Frameworks

CommonLoopUtils

CLU lets you write beautiful training loops in JAX.

Open Source

About

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.

Open Source Health

Not enough history
Stars
370
Forks
34
License
Apache-2.0
Last commit
3 months ago
Jupyter Notebook

Related Categories