Rollback PR #37936: since it breaks ragged_collective_test_gpu in JAX when machine have >2 GPUs.#41049
Open
copybara-service[bot] wants to merge 1 commit intomainfrom
Open
Rollback PR #37936: since it breaks ragged_collective_test_gpu in JAX when machine have >2 GPUs.#41049copybara-service[bot] wants to merge 1 commit intomainfrom
copybara-service[bot] wants to merge 1 commit intomainfrom
