transformers

[`distributed`] Add context parallelism: `DistributedConfig(cp_size=N)`

#50496

Open