Introducing TorchTPU: Native PyTorch on Google Cloud TPUs
Google Cloud Tech · 5:07 · Watch on YouTube
Torch TPU is Google’s open-source backend, built with Meta’s PyTorch team, that runs ordinary PyTorch on Cloud TPUs by treating the TPU as a native device. You keep the same model code, debug in eager mode, speed it u...