TorchShard: Slicing a PyTorch Tensor Into Parallel Shards. - View it on GitHub