PyPI page
Home page
Author:
ROCm JAX Devs
License:
Apache-2.0
Summary:
JAX Plugin for AMD GPUs (ROCm:7.2.0)
Latest version:
0.9.1.post3
Required dependencies:
jax-rocm7-pjrt
Optional dependencies:
amd_hipblas_instinct
|
amd_hipfft_instinct
|
amd_hipsolver_instinct
|
amd_hipsparse_instinct
|
amd_hipsparselt_instinct
|
amd_miopen_hip_instinct
|
amd_rccl_instinct
|
amd_rocm_device_libs_instinct
|
amd_rocm_hip_runtime_devel_instinct
|
amd_rocm_hip_runtime_instinct
|
amd_rocm_language_runtime_instinct
|
amd_rocm_llvm_instinct
Downloads last day:
151
Downloads last week:
2,295
Downloads last month:
3,670