# automatically generated by g-sorcery
# please do not edit this file

EAPI=8

REALNAME="${PN}"
LITERALNAME="${PN}"
REALVERSION="${PV}"
DIGEST_SOURCES="yes"
PYTHON_COMPAT=( python{3_11,3_12,3_13,3_14} )
DISTUTILS_USE_PEP517=standalone

inherit python-r1 gs-pypi

DESCRIPTION="Differentiate, compile, and transform Numpy code."

HOMEPAGE="https://github.com/jax-ml/jax"
LICENSE="Apache-2.0"
SRC_URI="https://files.pythonhosted.org/packages/source/${REALNAME::1}/${REALNAME}/${REALNAME}-${REALVERSION}.tar.gz"
SOURCEFILE="${REALNAME}-${REALVERSION}.tar.gz"
RESTRICT="test"

SLOT="0"
KEYWORDS="~amd64 ~x86"

IUSE="ci cuda cuda12 cuda12-local cuda13 cuda13-local k8s minimum-jaxlib rocm tpu xprof"
DEPENDENCIES="dev-python/jaxlib[${PYTHON_USEDEP}]
	>=dev-python/ml-dtypes-0.5.0[${PYTHON_USEDEP}]
	>=dev-python/numpy-2.0[${PYTHON_USEDEP}]
	dev-python/opt-einsum[${PYTHON_USEDEP}]
	>=dev-python/scipy-1.13[${PYTHON_USEDEP}]
	minimum-jaxlib? ( dev-python/jaxlib[${PYTHON_USEDEP}] )
	ci? ( dev-python/jaxlib[${PYTHON_USEDEP}] )
	tpu? ( dev-python/jaxlib[${PYTHON_USEDEP}] )
	tpu? ( dev-python/libtpu[${PYTHON_USEDEP}] )
	tpu? ( dev-python/requests[${PYTHON_USEDEP}] )
	cuda? ( dev-python/jaxlib[${PYTHON_USEDEP}] )
	cuda? ( dev-python/jax-cuda12-plugin[${PYTHON_USEDEP}] )
	cuda12? ( dev-python/jaxlib[${PYTHON_USEDEP}] )
	cuda12? ( dev-python/jax-cuda12-plugin[${PYTHON_USEDEP}] )
	cuda13? ( dev-python/jaxlib[${PYTHON_USEDEP}] )
	cuda13? ( dev-python/jax-cuda13-plugin[${PYTHON_USEDEP}] )
	cuda12-local? ( dev-python/jaxlib[${PYTHON_USEDEP}] )
	cuda12-local? ( dev-python/jax-cuda12-plugin[${PYTHON_USEDEP}] )
	cuda13-local? ( dev-python/jaxlib[${PYTHON_USEDEP}] )
	cuda13-local? ( dev-python/jax-cuda13-plugin[${PYTHON_USEDEP}] )
	rocm? ( dev-python/jaxlib[${PYTHON_USEDEP}] )
	rocm? ( dev-python/jax-rocm7-plugin[${PYTHON_USEDEP}] )
	k8s? ( dev-python/kubernetes[${PYTHON_USEDEP}] )
	xprof? ( dev-python/xprof[${PYTHON_USEDEP}] )"
BDEPEND="${DEPENDENCIES}"
RDEPEND="${DEPENDENCIES}"