jax
3eb27d36 - [CI] Skip Pallas test_fusible_shard_map_jit_grad on TPU7x.

Commit

51 days ago

[CI] Skip Pallas test_fusible_shard_map_jit_grad on TPU7x. PiperOrigin-RevId: 955951922

References

Author

belitskiy

Committer

Google-ML-Automation

Parents