跳到正文
原文
Hugging Face Blog·· 20 小时前精选AI 评分87

Hugging Face 发布 TorchTPU,让 PyTorch 原生运行在 TPU 上

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

Hugging Face 发布 TorchTPU,将 TPU 作为 PyTorch 的原生后端,使 Transformers、Diffusers 和 Kernels 能直接在 TPU 上运行。

推荐理由

Hugging Face 给出了 TorchTPU 的安装方式、库级接入和实测数据,展示 PyTorch 工作流迁移到 TPU 的具体路径。

来源:Hugging Face Blog · huggingface.co