PyPI Stats

Search

All packages
Top packages

Track packages

flax


PyPI page
Home page
Author: Flax team
License:
Summary: Flax: A neural network library for JAX designed for flexibility
Latest version: 0.6.0
Required dependencies: jax | matplotlib | msgpack | numpy | optax | pyyaml | rich | typing-extensions
Optional dependencies: atari-py | clu | gym | jaxlib | jraph | ml-collections | opencv-python | pytest | pytest-cov | pytest-custom-exit-code | pytest-xdist | pytype | sentencepiece | svn | tensorflow | tensorflow-datasets | tensorflow-text | torch

Downloads last day: 14,081
Downloads last week: 69,340
Downloads last month: 323,513