PyPI page
Home page
Author:
Common Loop Utils Authors
License:
Apache 2.0
Summary:
Set of libraries for ML training loops in JAX.
Latest version:
0.0.12
Required dependencies:
absl-py
|
etils
|
flax
|
jax
|
jaxlib
|
ml-collections
|
numpy
|
packaging
|
typing-extensions
|
wrapt
Optional dependencies:
pytest
|
tensorflow
|
tensorflow-datasets
|
torch
Downloads last day:
6,637
Downloads last week:
36,873
Downloads last month:
154,714