PyPI page
Author:
Flax team
Summary:
Flax: A neural network library for JAX designed for flexibility
Latest version:
0.12.10
Required dependencies:
jax
|
msgpack
|
numpy
|
optax
|
orbax-checkpoint
|
pyyaml
|
rich
|
tensorstore
|
treescope
|
typing-extensions
Optional dependencies:
ale-py
|
cloudpickle
|
clu
|
dm-haiku
|
docutils
|
einops
|
grain
|
gymnasium
|
ipykernel
|
ipython-genutils
|
ipywidgets
|
jaxlib
|
jaxtyping
|
jraph
|
jupytext
|
kagglehub
|
keras
|
matplotlib
|
ml-collections
|
mypy
|
myst-nb
|
nanobind
|
nbstripout
|
opencv-python
|
orbax-export
|
pre-commit
|
protobuf
|
pygments
|
pytest
|
pytest-cov
|
pytest-custom-exit-code
|
pytest-xdist
|
pytype
|
recommonmark
|
scikit-build-core
|
scikit-learn
|
sentencepiece
|
sphinx
|
sphinx-book-theme
|
sphinx-design
|
tensorboard
|
tensorflow
|
tensorflow-datasets
|
tensorflow-text
|
torch
|
tqdm
|
treescope
Downloads last day:
98,619
Downloads last week:
743,790
Downloads last month:
2,214,985
7-day smoothing