Run PyTorch on TPU and GPU without changing code
WIth PyTorch Lightning, you can run your PyTorch code on TPUs and GPUs without changing a single line of code. Simply organize your PyTorch code in the Lightning format and train!
https://github.com/PyTorchLightning/pytorch-lightning