Add TPU support
Although still experimental, we should add the according support for it. Likely as a distributed backend flag?
I'm not familiar with the TPU APIs but I assume there is a DP and DDP version? So, something like
Trainer(distributed_backend='tpu_dp')
Trainer(distributed_backend='tpu_ddp')