jax-rocm60-pjrt

PyPI page
Home page
Author: Ruturaj4
License: Apache-2.0
Summary: JAX XLA PJRT Plugin for AMD GPUs (ROCm:unknown)
Latest version: 0.5.0


Downloads last day: 256
Downloads last week: 316
Downloads last month: 770