TorchShard
https://medium.com/pytorch/torchshard-a31fcbfdc354
pytorch
,
model parallel
,
memory optimization