Add collective all_gather method to unofficial python interface.

Initial implementation is by ring-algorithm.
Intended use is via distribution strategies.

The implementation of all-gather is very similar to that of all-reduce,
so common code is factored out into ring_alg.{h,cc}.

PiperOrigin-RevId: 233775623