xla

1f0e972a - Support SPMD through the xla:// init_method (#5706)

Support SPMD through the xla:// init_method (#5706) * Support SPMD through the xla:// init_method * Ensure compatibility with multithreading * Add get_master_ip to runtime * Use for loop instead of iterator