PyTorch：Blog·· 14 天前精选AI 评分74
Google 与 Meta 发布 TorchTPU：PyTorch 原生 TPU 后端
Introducing TorchTPU
AI 导读
Google 与 Meta 联合开源 TorchTPU，为 Google TPU 提供原生 PyTorch 后端，模型只需把 .to("cuda") 改成 .to("tpu") 即可开箱运行，无需 JAX 依赖。
推荐理由
原文详述架构、精度模式与多个训练推理案例，读者可据此评估把 PyTorch 工作流迁移到 TPU 的成本。
来源：PyTorch：Blog · pytorch.org