PyPI page
Home page
Author:
None
License:
Apache License
Version 2.0, January 2004
http://www.apache.org/licenses/
TERMS AND CONDITIONS FOR USE, REPRODUCTI...
Summary:
None
Latest version:
2025.10.28
Required dependencies:
chex
|
flax
|
grain
|
jax
|
ml_dtypes
|
optax
|
orbax-checkpoint
|
orbax-export
Optional dependencies:
pytest
|
pytest-xdist
|
tensorflow
|
tensorflow_datasets
Downloads last day:
521
Downloads last week:
2,109
Downloads last month:
17,569