jax
b4e24590 - [Mosaic GPU] Add support for collective async copies.

Commit

138 days ago

[Mosaic GPU] Add support for collective async copies. PiperOrigin-RevId: 819065278

References

Author

allanrenucci

Committer

Google-ML-Automation

Parents