Pytorch XLA is not at all working on colab or kaggle kernels

Hi Everyone,
I am new to PyTorch and started using it recently for training on GPUs and TPUs.
I am not able to run any of the pytorch NN models on kaggle kernels and Google Colab.
Even the Notebook given on the Pytorch XLA Documentation doesnt work on Colab it just keeps on running for ever
and never shows an result.
Notebook: AlexNet on MNIST multicore
I even tried to Run on Single Core but speed was unbearable.
Even After This I tried using Pytorch Lightning Same Result It keeps on Running and never stops.
Without Any Error.
Can Anyone help or Guide What am I doing Wrong.
Thanks.