Weirdly enough, the notebook example is working for me with the normal RAM TPU from colab (if I change the pytorch version to !pip3 install torch==1.9 from the new colab default torch==1.10).
But I get the same SIGSEGV error, when changing to the high-RAM TPU from colab. the exact same code works with the low-ram TPU, does does not work with the high-ram TPU. probably an issue with google colab’s setup in the background …
Colab with traceback here: simple_nlp_example.ipynb - Google Drive