Machine Translation TPU Script

I am getting connection errors (cannot connect to tensorflow-mesh master) (Traceback is huge).
when I try to run
https://cloud.google.com/tpu/docs/tutorials/transformer-pytorch has anyone run this or any torch+machine translation +TPU script sucessfully?