xla

0f9bff34 - isolate the PJRT dependency between torch_xla and torch_xla2 (#7144)