在 Google 规模的 TPU 上本机运行 PyTorch
TorchTPU 是 Google TPU 的 PyTorch 原生后端。以最少的代码更改运行现有的 PyTorch 工作负载，通过 Fused Eager 模式获得 50-100% 以上的速度增益，并扩展到 100K 以上的芯片集群 - 无需静态图编译即可开始
还没有人点赞
