dev-python/flax
Neural network library and ecosystem for JAX
-
flax-0.12.6~amd64python_targets_python3_12 python_targets_python3_13
View
Download
Browse License: Apache-2.0 Overlay: pypi-sci -
flax-0.8.2amd64 ~arm64 x86python_targets_python3_12 python_targets_python3_13 python_targets_python3_14
View
Download
Browse License: Apache-2.0 Overlay: blshkv-overlay
Runtime Dependencies
flax-0.12.6
~dev-python/jax-0.9.2[rocm,python_targets_python3_12
(-)
?
,python_targets_python3_13
(-)
?
]
>=dev-python/numpy-1.23.2[python_targets_python3_12
(-)
?
,python_targets_python3_13
(-)
?
]
dev-python/msgpack[python_targets_python3_12
(-)
?
,python_targets_python3_13
(-)
?
]
dev-python/optax[python_targets_python3_12
(-)
?
,python_targets_python3_13
(-)
?
]
dev-python/orbax-checkpoint[python_targets_python3_12
(-)
?
,python_targets_python3_13
(-)
?
]
>=dev-python/orbax-export-0.0.8[python_targets_python3_12
(-)
?
,python_targets_python3_13
(-)
?
]
dev-python/tensorstore[python_targets_python3_12
(-)
?
,python_targets_python3_13
(-)
?
]
>=dev-python/rich-11.1[python_targets_python3_12
(-)
?
,python_targets_python3_13
(-)
?
]
>=dev-python/typing-extensions-4.2[python_targets_python3_12
(-)
?
,python_targets_python3_13
(-)
?
]
>=dev-python/pyyaml-5.4.1[python_targets_python3_12
(-)
?
,python_targets_python3_13
(-)
?
]
>=dev-python/treescope-0.1.7[python_targets_python3_12
(-)
?
,python_targets_python3_13
(-)
?
]
python_targets_python3_12?
( dev-lang/python:3.12 )
python_targets_python3_13?
( dev-lang/python:3.13 )
flax-0.8.2
python_targets_python3_12?
( dev-lang/python:3.12 )
python_targets_python3_13?
( dev-lang/python:3.13 )
python_targets_python3_14?
( dev-lang/python:3.14 )

