Add `WaitAtBarrier` with subset of nodes to `DistributedRuntimeCoordinationServiceClient` and plumb that up to the JAX API layer.
Add `WaitAtBarrier` with subset of nodes to `DistributedRuntimeCoordinationServiceClient` and plumb that up to the JAX API layer.
