jax-rocm7-plugin 0.11.2

Author: ROCm JAX Devs

Requires Python: >=3.12

jax_rocm7_plugin-0.11.2-cp312-cp312-manylinux_2_27_x86_64.whl (4.1MiB)

jax_rocm7_plugin-0.11.2-cp313-cp313-manylinux_2_27_x86_64.whl (4.1MiB)

jax_rocm7_plugin-0.11.2-cp314-cp314-manylinux_2_27_x86_64.whl (4.1MiB)

jax_rocm7_plugin-0.11.2-cp315-cp315-manylinux_2_27_x86_64.whl (4.1MiB)

Dependencies:

jax-rocm7-pjrt (==0.11.2)

amd_rocm_hip_runtime_devel_instinct extra == "with-rocm"

amd_rocm_hip_runtime_instinct extra == "with-rocm"

amd_hipblas_instinct extra == "with-rocm"

amd_hipsparse_instinct extra == "with-rocm"

amd_hipsolver_instinct extra == "with-rocm"

amd_miopen_hip_instinct extra == "with-rocm"

amd_rocm_llvm_instinct extra == "with-rocm"

amd_rocm_language_runtime_instinct extra == "with-rocm"

amd_rccl_instinct extra == "with-rocm"

amd_hipfft_instinct extra == "with-rocm"

amd_rocm_device_libs_instinct extra == "with-rocm"

amd_hipsparselt_instinct extra == "with-rocm"