Running torchscript on multiple GPUs

TorchScript is in maintenance mode and not ins active development anymore so I would recommend switching to the torch.compile stack.