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

AratiGanesh wants to merge 1 commit intorocm-jaxlib-v0.9.0ROCm/jax:rocm-jaxlib-v0.9.0from

araganes/skip-pmap-sharding-errorROCm/jax:araganes/skip-pmap-sharding-errorCopy head branch name to clipboard