PyPI page
Home page
Author:
ROCm JAX Devs
License:
Apache-2.0
Summary:
JAX Plugin for AMD GPUs (ROCm:7.14.0)
Latest version:
0.12.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:
125
Downloads last week:
348
Downloads last month:
1,484
7-day smoothing