Comment by Akababa

I'm getting OOM errors on colab after the first 10-ish actions :(

EDIT: Any chance you can support TPU on Colab?