Make a PyTorch model tensor-parallel in one line of code
github.com