Fix incorrect src argument in broadcast_params function

[Yuxin-CV]

Corrects the broadcast_params function to use the first rank of the process group as the source for parameter broadcasting. Previously, the function erroneously passed all ranks of the data_parallel_group to the src parameter, which expects a single integer. This update ensures that only the first rank is used as the source, aligning with expected torch.distributed.broadcast usage.