dev-python/flax - 0.12.6 (pypi-sci)

Search

Install

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

Package Information

Description:
Neural network library and ecosystem for JAX
Homepage:
https://github.com/google/flax
License:
Apache-2.0

Ebuild Details

Version EAPI Keywords Slot
0.12.6 8 ~amd64 0
View Raw Ebuild
# 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}]"

Inherited Eclasses

Dependencies

RDEPEND

	~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[]

BDEPEND

dev-python/setuptools-scm[]