Add comm and distributed kernels

[xuzhao9]

PyTorch Symmetric Memory: https://github.com/meta-pytorch/kraken
torchcomms/NCCL-X: https://github.com/meta-pytorch/torchcomms
UCCL: https://github.com/uccl-project/uccl