jax
Skip test_pmap on ROCm due to IndivisibleError with new pmap SPMD tiling
#725
Open
Open