google-pytorch / torch_tpu

Provides a PyTorch backend that runs PyTorch models and operations on Google's Tensor Processing Units (TPUs).