Hi!, I know the code is made in order to use Multi GPU or Single GPU. But maybe it has been discussed to extend it for TPU? (And use free capabilities like Google Colab).

Thanks!