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 5 is my colab notebook. Please help.