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.8.3
Required dependencies: clu | jax | msgpack | numpy | optax | orbax-checkpoint | penzai | pyyaml | rich | tensorstore | typing-extensions
Optional dependencies: black | clu | einops | gymnasium | jaxlib | jraph | matplotlib | ml-collections | mypy | nbstripout | opencv-python | pytest | pytest-cov | pytest-custom-exit-code | pytest-xdist | pytype | sentencepiece | tensorflow | tensorflow-datasets | tensorflow-text | torch

Downloads last day: 132,923
Downloads last week: 908,679
Downloads last month: 4,254,834