# 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=wheel

inherit python-r1 gs-pypi

DESCRIPTION="JAX Plugin for NVIDIA GPUs [wheel]"

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

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

IUSE="with-cuda"
DEPENDENCIES="dev-python/jax-cuda12-pjrt[${PYTHON_USEDEP}]
	with-cuda? ( dev-python/nvidia-cublas-cu12[${PYTHON_USEDEP}] )
	with-cuda? ( dev-python/nvidia-cuda-cupti-cu12[${PYTHON_USEDEP}] )
	with-cuda? ( dev-python/nvidia-cuda-nvcc-cu12[${PYTHON_USEDEP}] )
	with-cuda? ( dev-python/nvidia-cuda-runtime-cu12[${PYTHON_USEDEP}] )
	with-cuda? ( dev-python/nvidia-cudnn-cu12[${PYTHON_USEDEP}] )
	with-cuda? ( dev-python/nvidia-cufft-cu12[${PYTHON_USEDEP}] )
	with-cuda? ( dev-python/nvidia-cusolver-cu12[${PYTHON_USEDEP}] )
	with-cuda? ( dev-python/nvidia-cusparse-cu12[${PYTHON_USEDEP}] )
	with-cuda? ( dev-python/nvidia-nccl-cu12[${PYTHON_USEDEP}] )
	with-cuda? ( dev-python/nvidia-nvjitlink-cu12[${PYTHON_USEDEP}] )
	with-cuda? ( dev-python/nvidia-cuda-nvrtc-cu12[${PYTHON_USEDEP}] )
	with-cuda? ( dev-python/nvidia-nvshmem-cu12[${PYTHON_USEDEP}] )
	virtual/allow-pypi-wheels"
BDEPEND="${DEPENDENCIES}"
RDEPEND="${DEPENDENCIES}"