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.3.5
Required dependencies: dataclasses | jax | matplotlib | msgpack | numpy | optax
Optional dependencies: atari-py | clu | gym | jaxlib | ml-collections | opencv-python | pytest | pytest-cov | pytest-xdist | pytype | sentencepiece | svn | tensorflow | tensorflow-cpu | tensorflow-datasets | tensorflow-text

Downloads last day: 24,903
Downloads last week: 126,520
Downloads last month: 494,429