jax-rocm7-pjrt

PyPI page
Home page
Author: ROCm JAX Devs
License: Apache-2.0
Summary: JAX XLA PJRT Plugin for AMD GPUs (ROCm: Legacy ROCm 7.2.0)
Latest version: 0.9.2.post1


Downloads last day: 206
Downloads last week: 3,212
Downloads last month: 10,524