跳到正文
Hugging Face Blog·· 2 天前精选AI 评分79

Hugging Face 发布 TorchTPU 原生 PyTorch 后端

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

Hugging Face 发布 TorchTPU,将 TPU 集成到 PyTorch,并让 Transformers、Diffusers 和 Kernels 可原生运行在 TPU 上。

推荐理由

TorchTPU把TPU纳入PyTorch原生后端,并披露Transformers、Diffusers与Kernels的运行方式及测试进度,便于评估迁移成本与成熟度。

来源:Hugging Face Blog · huggingface.co