Code and visualisations about PyTorch distributed collective communication - View it on GitHub