Install this version:
emerge -a =dev-python/flax-0.12.6
If this version is masked, you can unmask it using the autounmask tool or standard emerge options:
autounmask =dev-python/flax-0.12.6
Or alternatively:
emerge --autounmask-write -a =dev-python/flax-0.12.6
| Version | EAPI | Keywords | Slot |
|---|---|---|---|
| 0.12.6 | 8 | ~amd64 | 0 |
# Copyright 1999-2026 Gentoo Authors
# Distributed under the terms of the GNU General Public License v2
EAPI=8
DISTUTILS_USE_PEP517=setuptools
PYTHON_COMPAT=( python3_{12..13} )
inherit distutils-r1 pypi
DESCRIPTION="Neural network library and ecosystem for JAX"
HOMEPAGE="https://github.com/google/flax"
LICENSE="Apache-2.0"
SLOT="0"
KEYWORDS="~amd64"
RDEPEND="
~dev-python/jax-0.9.2[rocm,${PYTHON_USEDEP}]
>=dev-python/numpy-1.23.2[${PYTHON_USEDEP}]
dev-python/msgpack[${PYTHON_USEDEP}]
dev-python/optax[${PYTHON_USEDEP}]
dev-python/orbax-checkpoint[${PYTHON_USEDEP}]
>=dev-python/orbax-export-0.0.8[${PYTHON_USEDEP}]
dev-python/tensorstore[${PYTHON_USEDEP}]
>=dev-python/rich-11.1[${PYTHON_USEDEP}]
>=dev-python/typing-extensions-4.2[${PYTHON_USEDEP}]
>=dev-python/pyyaml-5.4.1[${PYTHON_USEDEP}]
>=dev-python/treescope-0.1.7[${PYTHON_USEDEP}]
"
BDEPEND="dev-python/setuptools-scm[${PYTHON_USEDEP}]"
~dev-python/jax-0.9.2[rocm,] >=dev-python/numpy-1.23.2[] dev-python/msgpack[] dev-python/optax[] dev-python/orbax-checkpoint[] >=dev-python/orbax-export-0.0.8[] dev-python/tensorstore[] >=dev-python/rich-11.1[] >=dev-python/typing-extensions-4.2[] >=dev-python/pyyaml-5.4.1[] >=dev-python/treescope-0.1.7[]
dev-python/setuptools-scm[]