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

Hugging Face 发布 TorchTPU 原生 PyTorch 后端

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

Hugging Face 发布 TorchTPU,将 TPU 作为 PyTorch 原生后端,并让 Transformers、Diffusers 和 Kernels 原生支持 TPU。

推荐理由

TorchTPU 让 PyTorch 与 Hugging Face 工作流原生接入 TPU,并给出安装、设备映射及性能数据参考。

来源:Hugging Face Blog · huggingface.co