Make a PyTorch model tensor-parallel in one line of code | Heykuki News