Published 22 days ago
JAX XLA PJRT Plugin for AMD GPUs (ROCm:unknown)
pip install jax-rocm60-pjrt
[email protected]
