Hugging Face Blog·· 17 小时前精选AI 评分83
Hugging Face 推出 TorchTPU,为 PyTorch 原生支持 TPU
Hugging Face ❤️ TorchTPU: TPUs, Natively
AI 导读
Hugging Face 推出 TorchTPU,为 PyTorch 增加原生 TPU 后端,使 Transformers、Diffusers 和 Kernels 可在 TPU 上运行。用户可用常规 PyTorch 代码加载模型,并使用 torch.compile、分布式训练和 TPU kernels;测试中已运行的 TPU 测试有 96.5% 通过,34 个问题中 19 个已修复。
推荐理由
文章说明 TorchTPU 如何把 TPU 接入 PyTorch 及 Hugging Face 生态,并提供性能数据与测试通过率,便于了解其支持范围和当前成熟度。
来源:Hugging Face Blog · huggingface.co