jax
Backporting `device test` unit tests for ROCm from JAX v0.9.0 to v0.8.2.
#706
Merged
Merged