xla

3078d163 - [torch_xla2] Functional collectives (#7580)