collective_reduce_scatter.py 3.0 KB