PyPI page
Home page
Author:
Ruturaj4
License:
Apache-2.0
Summary:
JAX Plugin for AMD GPUs (ROCm:unknown)
Latest version:
0.7.1
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:
5
Downloads last week:
80
Downloads last month:
318