PyPI page
Home page
Author:
ROCm JAX Devs
License:
Apache-2.0
Summary:
JAX Plugin for AMD GPUs (ROCm:10.0.0)
Latest version:
0.11.2
Required dependencies:
jax-rocm10-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:
183
Downloads last week:
1,098
Downloads last month:
3,783
7-day smoothing