PyPI page
Author:
jax-ai-stack authors
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:
16
Downloads last week:
145
Downloads last month:
358
7-day smoothing