flax

PyPI page
Home page
Author: None
Summary: Flax: A neural network library for JAX designed for flexibility
Latest version: 0.12.4
Required dependencies: ale-py | clu | gymnasium | jax | msgpack | numpy | optax | orbax-checkpoint | orbax-export | protobuf | pyyaml | rich | tensorflow | tensorflow_text | tensorstore | treescope | typing_extensions
Optional dependencies: chex | cloudpickle | clu | dm-haiku | docutils | einops | ipykernel | ipython_genutils | ipywidgets | jaxlib | jaxtyping | jraph | jupytext | kagglehub | keras | matplotlib | ml-collections | ml_collections | mypy | myst_nb | nanobind | nbstripout | opencv-python | pre-commit | pygments | pytest | pytest-cov | pytest-custom_exit_code | pytest-xdist | pytype | recommonmark | scikit-build-core | scikit-learn | sentencepiece | sphinx | sphinx-book-theme | sphinx-design | tensorflow_datasets | torch | tqdm

Downloads last day: 142,330
Downloads last week: 1,479,023
Downloads last month: 4,127,473