Running PyTorch Natively on TPUs at Google Scale
TorchTPU is Google's PyTorch-native backend for TPUs. Run existing PyTorch workloads with minimal code changes, get 50-100%+ speed gains with Fused Eager mode, and scale to 100K+ chip clusters — no static graph compilation required to get started