jax
95c4ba96 - [JAX] Use first process id instead of process 0 to share multi-host data.

Commit

1 year ago

[JAX] Use first process id instead of process 0 to share multi-host data. PiperOrigin-RevId: 633526220

References

Author

a-googler

Committer

a-googler

Parents