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