TorchTPU logo

TorchTPU

Running PyTorch Natively on TPUs at Google Scale

Artificial Intelligence Developer Tools Open Source

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

投票数: 77
← 投稿一覧に戻る