PyPI page
Home page
Author:
None
Summary:
Flax: A neural network library for JAX designed for flexibility
Latest version:
0.8.5
Required dependencies:
clu
|
jax
|
msgpack
|
numpy
|
optax
|
orbax-checkpoint
|
penzai
|
pyyaml
|
rich
|
tensorstore
|
typing-extensions
Optional dependencies:
black
|
clu
|
einops
|
gymnasium
|
jaxlib
|
jaxtyping
|
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:
50,869
Downloads last week:
226,123
Downloads last month:
1,113,207