pytorch code examples for measuring the performance of collective communication calls in AI workloads - View it on GitHub