gpo.zugaina.org

Search Portage & Overlays:

dev-python/jax-rocm7-plugin

ROCm plugin for JAX

Screenshots

  • jax-rocm7-plugin-0.11.1
    ~amd64
    python_targets_python3_13 debug

    View      Download      Browse     License: Apache-2.0   
    Overlay: pypi-sci
  • jax-rocm7-plugin-0.9.2
    ~amd64
    python_targets_python3_12 python_targets_python3_13 debug

    View      Download      Browse     License: Apache-2.0   
    Overlay: pypi-sci

Runtime Dependencies

jax-rocm7-plugin-0.11.1


~dev-python/jax-rocm7-pjrt-0.11.1[python_targets_python3_13
      (-)
?
]
=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

python_targets_python3_13?
      ( dev-lang/python:3.13 )

jax-rocm7-plugin-0.9.2



~dev-python/jax-rocm7-pjrt-0.9.2[python_targets_python3_12
      (-)
?
,python_targets_python3_13
      (-)
?
]
=dev-build/rocm-cmake-7.1.0*
=dev-libs/rocm-comgr-7.1.0*
=dev-libs/rocm-core-7.1.0*
=dev-libs/rocm-device-libs-7.1.0*
=dev-libs/rocr-runtime-7.1.0*
=dev-util/hip-7.1.0*
=dev-util/hipcc-7.1.0*
=dev-util/hipify-clang-7.1.0*
=dev-util/rocm-smi-7.1.0*
=dev-util/rocminfo-7.1.0*
=dev-util/roctracer-7.1.0*
=dev-libs/rccl-7.1.0*
=sci-libs/hipBLAS-7.1.0*
=sci-libs/hipBLASLt-7.1.0*
=sci-libs/hipFFT-7.1.0*
=sci-libs/hipRAND-7.1.0*
=sci-libs/hipSOLVER-7.1.0*
=sci-libs/hipSPARSE-7.1.0*
=sci-libs/miopen-7.1.0*
=sci-libs/rocBLAS-7.1.0*
=sci-libs/rocFFT-7.1.0*
=sci-libs/rocRAND-7.1.0*
=sci-libs/rocSOLVER-7.1.0*
=sci-libs/rocSPARSE-7.1.0*
llvm-core/clang:18
llvm-core/lld:18

python_targets_python3_12?
      ( dev-lang/python:3.12 )

python_targets_python3_13?
      ( dev-lang/python:3.13 )