Place it in a torch_xla.backends module,

document that it should be moved to torch.backends.xla eventually.