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