Suppose that I have a big model that cannot fit into one gpu, so I have to split the model to different gpus. I’m wondering whether there’s a way to make Pytorch take multiple gpu as a single one, so that we don’t have to split model manually.