xla
Implement `xm.rendezvous` with XLA collective communication
#4181

Merged

Commits

Implement pjrt.rendezvous with XLA collective ops.

will-cromar committed 3 years ago
Update tests.

will-cromar committed 3 years ago
Formatting

will-cromar committed 3 years ago
Handle some edge cases better

will-cromar committed 3 years ago
formatting

will-cromar committed 3 years ago
Add mesh service test to CI

will-cromar committed 3 years ago