xla

8e634d0e - Support eager mode for multi-process training (#7327)