Hi sgugger, sorry for the inconvenience.
Can you delete all the code in this notebook and replace the code with the SIGKILL one?
https://colab.research.google.com/github/pytorch/xla/blob/master/contrib/colab/mnist-training.ipynb

This notebook is from pytorchXLA demo and have an instance with more ram 35gb so it should work (for debugging purpose). The standard colab one only have 15gb.
Thanks