How to make it work with TPU?

[mulgray]

Thank you for releasing the great project.
I'd like to train on TPU(v3) on Google Cloud, how can I change the code to make it work?

[ivanvovk]

@mulgray To be honest, I've never run PyTorch with TPU before, but I know official PyTorch XLA framework may help to connect these things together: https://github.com/pytorch/xla

[mulgray]

Thank you for your reply. I will give it a try.