I wanted to try the PyTorch lightning tutorial on MNIST on colab TPU. But, after running it with some modifications, I am getting
Exception in device=TPU:0: module ‘torch_xla.core.xla_model’ has no attribute ‘rendezvous’
Here is my colab notebook. Please help.