Adds support for allgather_into_tensor_coalesced and reduce_scatter_tensor_coalesced
#350
| Job | Run time |
|---|---|
| 2m 58s | |
| 2m 58s |
allgather_into_tensor_coalesced and reduce_scatter_tensor_coalesced
#350
| Job | Run time |
|---|---|
| 2m 58s | |
| 2m 58s |