jax
bff24bdd - Add axis_index_groups support to all_gather. (#4194)

Commit

5 years ago

Add axis_index_groups support to all_gather. (#4194)

References

#4194 - Add axis_index_groups support to all_gather.

Author

inoryy

inoryy

Parents

Loading