xla

d228fc96 - Remove usage of torch_xla_py (#1130)