PyPI Stats

Search

All packages
Top packages

Track packages

flax


PyPI page
Home page
Author: None
Summary: Flax: A neural network library for JAX designed for flexibility
Latest version: 0.10.1
Required dependencies: clu | jax | msgpack | numpy | optax | orbax-checkpoint | pyyaml | rich | tensorflow-text | tensorstore | treescope | typing-extensions
Optional dependencies: cloudpickle | clu | dm-haiku | docutils | einops | gymnasium | ipykernel | ipython-genutils | ipywidgets | jaxlib | jaxtyping | jraph | jupytext | kagglehub | matplotlib | ml-collections | mypy | myst-nb | nbstripout | opencv-python | pre-commit | pygments | pytest | pytest-cov | pytest-custom-exit-code | pytest-xdist | pytype | recommonmark | scikit-learn | sentencepiece | sphinx | sphinx-book-theme | sphinx-design | tensorflow | tensorflow-datasets | torch

Downloads last day: 33,748
Downloads last week: 204,822
Downloads last month: 894,091