flax

PyPI page
Home page
Author: None
Summary: Flax: A neural network library for JAX designed for flexibility
Latest version: 0.12.1
Required dependencies: clu | jax | msgpack | numpy | optax | orbax-checkpoint | protobuf | pyyaml | rich | tensorflow | tensorflow_text | tensorstore | treescope | typing_extensions
Optional dependencies: ale-py | cloudpickle | clu | dm-haiku | docutils | einops | gymnasium | ipykernel | ipython_genutils | ipywidgets | jaxlib | jaxtyping | jraph | jupytext | kagglehub | 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

Downloads last day: 109,744
Downloads last week: 684,852
Downloads last month: 2,909,418