Gentoo Packages

dev-python/jax - 0.9.0 (pypi)

Search

Package Information

Description:
Differentiate, compile, and transform Numpy code.
Homepage:
https://github.com/jax-ml/jax
License:
Apache-2.0

Ebuild Details

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

USE Flags

ci
cuda
cuda12
cuda12-local
cuda13
cuda13-local
k8s
minimum-jaxlib
rocm
tpu
xprof

Dependencies

RDEPEND

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

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}] )

Manifest for 0.9.0

Type File Size Source URLs
DIST jax-0.9.0.tar.gz 2534495 bytes https://files.pythonhosted.org/packages/source/${REALNAME::1}/jax/jax-0.9.0.tar.gz