Install this version:
emerge -a =dev-python/jax-rocm7-pjrt-0.11.1If this version is masked, you can unmask it using the autounmask tool or standard emerge options:
autounmask =dev-python/jax-rocm7-pjrt-0.11.1Or alternatively:
emerge --autounmask-write -a =dev-python/jax-rocm7-pjrt-0.11.1| Version | EAPI | Keywords | Slot |
|---|---|---|---|
| 0.11.1 | 8 | ~amd64 | 0 |
# Copyright 1999-2026 Gentoo Authors
# Distributed under the terms of the GNU General Public License v2
EAPI=8
PYPI_PN=${PN}
DISTUTILS_EXT=1
DISTUTILS_USE_PEP517=standalone
PYTHON_COMPAT=( python3_13 )
inherit distutils-r1 pypi
DESCRIPTION="ROCm PJRT runtime for JAX"
HOMEPAGE="https://github.com/ROCm/rocm-jax"
SRC_URI="
amd64? (
$(pypi_wheel_url "${PYPI_PN}" "${PV}" py3 none-manylinux_2_27_x86_64)
)
"
S=${WORKDIR}
LICENSE="Apache-2.0"
SLOT="0"
KEYWORDS="~amd64"
RESTRICT="strip"
ROCM_VERSION=7.2.0
LLVM_SLOT=22
RDEPEND="
=dev-build/rocm-cmake-${ROCM_VERSION}*
=dev-libs/rocm-comgr-${ROCM_VERSION}*
=dev-libs/rocm-core-${ROCM_VERSION}*
=dev-libs/rocm-device-libs-${ROCM_VERSION}*
=dev-libs/rocr-runtime-${ROCM_VERSION}*
=dev-util/hip-${ROCM_VERSION}*
=dev-util/hipcc-${ROCM_VERSION}*
=dev-util/hipify-clang-${ROCM_VERSION}*
=dev-util/rocm-smi-${ROCM_VERSION}*
=dev-util/rocminfo-${ROCM_VERSION}*
=dev-util/roctracer-${ROCM_VERSION}*
=dev-libs/rccl-${ROCM_VERSION}*
=sci-libs/hipBLAS-${ROCM_VERSION}*
=sci-libs/hipBLASLt-${ROCM_VERSION}*
=sci-libs/hipFFT-${ROCM_VERSION}*
=sci-libs/hipRAND-${ROCM_VERSION}*
=sci-libs/hipSOLVER-${ROCM_VERSION}*
=sci-libs/hipSPARSE-${ROCM_VERSION}*
=sci-libs/miopen-${ROCM_VERSION}*
=sci-libs/rocBLAS-${ROCM_VERSION}*
=sci-libs/rocFFT-${ROCM_VERSION}*
=sci-libs/rocRAND-${ROCM_VERSION}*
=sci-libs/rocSOLVER-${ROCM_VERSION}*
=sci-libs/rocSPARSE-${ROCM_VERSION}*
llvm-core/clang:${LLVM_SLOT}
llvm-core/lld:${LLVM_SLOT}
"
QA_PREBUILT=(
"/usr/lib/python*/site-packages/jax_plugins/xla_rocm7/*.so"
)
python_compile() {
local wheel="$(pypi_wheel_name "${PYPI_PN}" "${PV}" py3 none-manylinux_2_27_x86_64)"
distutils_wheel_install "${BUILD_DIR}/install" "${DISTDIR}/${wheel}"
}
=dev-build/rocm-cmake-7.2.0* =dev-libs/rocm-comgr-7.2.0* =dev-libs/rocm-core-7.2.0* =dev-libs/rocm-device-libs-7.2.0* =dev-libs/rocr-runtime-7.2.0* =dev-util/hip-7.2.0* =dev-util/hipcc-7.2.0* =dev-util/hipify-clang-7.2.0* =dev-util/rocm-smi-7.2.0* =dev-util/rocminfo-7.2.0* =dev-util/roctracer-7.2.0* =dev-libs/rccl-7.2.0* =sci-libs/hipBLAS-7.2.0* =sci-libs/hipBLASLt-7.2.0* =sci-libs/hipFFT-7.2.0* =sci-libs/hipRAND-7.2.0* =sci-libs/hipSOLVER-7.2.0* =sci-libs/hipSPARSE-7.2.0* =sci-libs/miopen-7.2.0* =sci-libs/rocBLAS-7.2.0* =sci-libs/rocFFT-7.2.0* =sci-libs/rocRAND-7.2.0* =sci-libs/rocSOLVER-7.2.0* =sci-libs/rocSPARSE-7.2.0* llvm-core/clang:22 llvm-core/lld:22