Enabling PyTorch on XLA Devices (e.g. Google TPU) - View it on GitHub