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.10.0
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:
69
Downloads last week:
1,348
Downloads last month:
9,908