jax-rocm10-pjrt

PyPI page
Home page
Author: ROCm JAX Devs
License: Apache-2.0
Summary: JAX XLA PJRT Plugin for AMD GPUs (ROCm:10.0.0)
Latest version: 0.11.2


Downloads last day: 183
Downloads last week: 1,085
Downloads last month: 3,828
7-day smoothing