Skip to content

Merge branch 'master' into jax-tpu-multi-slice

b2bf780
Select commit
Loading
Failed to load commit list.
Merged

[Train] Add TPU multi-slice support to JaxTrainer #58629

Merge branch 'master' into jax-tpu-multi-slice
b2bf780
Select commit
Loading
Failed to load commit list.