DeepSpeed

Fix auto TP for duplicate modules with different gems

#2784

Merged