jax

0fb48a15 - Include prebuilt rocm plugin pkgs as deps to tests