transformers

Simplify Tensor Parallel implementation with PyTorch TP

#34184

Merged