jaxlib-rocm (pypi): XLA library for JAX
Keywords: jax