jax-rocm60-plugin

PyPI page
Home page
Author: Ruturaj4
License: Apache-2.0
Summary: JAX Plugin for AMD GPUs (ROCm:unknown)
Latest version: 0.5.0
Required dependencies: jax-rocm60-pjrt

Downloads last day: 33
Downloads last week: 117
Downloads last month: 444