jax
b19be1aa - Move collective metadata initialization to the Initialize stage of the Mosaic GPU custom call.

Commit

241 days ago

Move collective metadata initialization to the Initialize stage of the Mosaic GPU custom call. PiperOrigin-RevId: 861057852

References

Author

Google-ML-Automation

Committer

Google-ML-Automation

Parents