cgarciae / ciclo

A functional training loops library for JAX

https://cgarciae.github.io/ciclo

cgarciae/ciclo Issues

Error importing ciclo
Updated 7 months ago4
Unexpected behaviour for reducing metrics in `flax/02_mnist_train_loop.py`
Updated 9 months ago
callbacks.checkpoints uses old flax.training.checkpoints API
Updated 9 months ago1
callbacks.early_stopping ignores min_delta and other thoughts
Updated 9 months ago1
inner_loop assumes the use of stateful metrics
Closed 9 months ago1
Surprising history.collect behavior with train_loop + test_step.
Updated 10 months ago2
Validation is only executed once in `02_mnist_train_loop.py` example
Closed a year ago2
Python 3.11 support
Closed a year ago2