PyPI page
Home page
Author:
ROCm JAX Devs
License:
Apache-2.0
Summary:
JAX Plugin for AMD GPUs (ROCm: Legacy ROCm 7.2.0)
Latest version:
0.9.2.post1
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:
378
Downloads last week:
2,896
Downloads last month:
10,244