openteamsinc
pypi/flax
Risk Profile
Package
Source
Risk Profile
Maturity:
Mature
Health:
Healthy
Legal:
Caution Needed
License includes a patent grant clause
Security:
Healthy
Package
pypi
flax
Version
0.10.7
Last Release Date
15 days ago
License
Apache
Dependencies
numpy
>=1.23.2
numpy
>=1.26.0
jax
>=0.6.0
msgpack
optax
orbax-checkpoint
tensorstore
rich
>=11.1
typing_extensions
>=4.2
PyYAML
>=5.4.1
treescope
>=0.1.7
all:
matplotlib
testing:
clu
clu
<=0.0.9
einops
gymnasium
[accept-rom-license,atari]
jaxlib
jaxtyping
jraph
>=0.0.6dev0
ml-collections
mypy
opencv-python
pytest
pytest-cov
pytest-custom_exit_code
pytest-xdist
pytype
sentencepiece
tensorflow_text
>=2.11.0
tensorflow_datasets
tensorflow
>=2.12.0
torch
treescope
>=0.1.1
cloudpickle
>=3.0.0
ale-py
>=0.10.2
docs:
sphinx
>=4.3.0
sphinx-book-theme
Pygments
>=2.6.1
ipykernel
myst_nb
nbstripout
recommonmark
ipython_genutils
sphinx-design
jupytext
==1.13.8
dm-haiku
docutils
matplotlib
scikit-learn
ml_collections
einops
kagglehub
>=0.3.3
ipywidgets
>=8.1.5
dev:
nanobind
>=2.5.0
pre-commit
>=3.8.0
scikit-build-core
[pyproject]
>=0.11.0
Source
Location
google/flax
Last Source Update
about 9 hours ago
Licenses
Apache License 2.0
Apache License 2.0
Distribution Destinations
pypi/flax
pypi/flaxlib