Tensor Parallelism .. py:module:: torch.distributed.tensor.parallel .. currentmodule:: torch.distributed.tensor.parallel