jax-rocm7-plugin 0.9.1.post3

Author: ROCm JAX Devs

Requires Python: >=3.11

jax_rocm7_plugin-0.9.1.post3-cp311-cp311-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl (7.5MiB)

jax_rocm7_plugin-0.9.1.post3-cp312-cp312-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl (7.5MiB)

jax_rocm7_plugin-0.9.1.post3-cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl (7.5MiB)

jax_rocm7_plugin-0.9.1.post3-cp314-cp314-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl (7.5MiB)

Dependencies:

jax-rocm7-pjrt (==0.9.1.*)

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"