jax-rocm7-plugin 0.10.0

Author: ROCm JAX Devs

Requires Python: >=3.11

jax_rocm7_plugin-0.10.0-cp311-cp311-manylinux_2_28_x86_64.whl (7.7MiB)

jax_rocm7_plugin-0.10.0-cp312-cp312-manylinux_2_28_x86_64.whl (7.7MiB)

jax_rocm7_plugin-0.10.0-cp313-cp313-manylinux_2_28_x86_64.whl (7.7MiB)

jax_rocm7_plugin-0.10.0-cp314-cp314-manylinux_2_28_x86_64.whl (7.7MiB)

Dependencies:

jax-rocm7-pjrt (==0.10.0)

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"