Install this version:
emerge -a =dev-python/jax-0.4.25
If this version is masked, you can unmask it using the autounmask tool or standard emerge options:
autounmask =dev-python/jax-0.4.25
Or alternatively:
emerge --autounmask-write -a =dev-python/jax-0.4.25
# Copyright 1999-2024 Gentoo Authors
# Distributed under the terms of the GNU General Public License v2
EAPI=8
DISTUTILS_USE_PEP517=setuptools
PYTHON_COMPAT=( python3_{12..14} )
inherit distutils-r1 pypi
DESCRIPTION="JAX: Autograd and XLA"
HOMEPAGE="https://pypi.org/project/jax/"
LICENSE=""
SLOT="0"
KEYWORDS="~amd64 ~arm64 ~x86"
RDEPEND=""
DEPEND="${RDEPEND}"
REQUIRED_USE="${PYTHON_REQUIRED_USE}"
RESTRICT="test"
#distutils_enable_tests pytest