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

Hugging Face 发布原生 PyTorch 后端 TorchTPU

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

Hugging Face 发布 TorchTPU,将 TPU 作为 PyTorch 原生后端,并让 Transformers、Diffusers 与 Kernels 可直接在 TPU 上运行。

推荐理由

TorchTPU 将 TPU 原生接入 PyTorch 与 Hugging Face 生态,并公开安装、调用和测试进展,便于评估迁移成本。

来源:Hugging Face Blog · huggingface.co