Hugging Face Blog·· 17 小时前精选AI 评分66
Hugging Face 宣布原生支持 TorchTPU 后端
Hugging Face ❤️ TorchTPU: TPUs, Natively
AI 导读
Hugging Face 宣布为 PyTorch 引入原生 TPU 支持的 TorchTPU 后端,让 TPU 在 Transformers 和 Diffusers 生态中成为一等设备。开发者只需将模型指定到 tpu 设备即可直接运行,无需再依赖 torch_xla 或 Optimum TPU。
推荐理由
原文给出了新后端在 Hugging Face 生态中的具体用法和安装方式,想在 TPU 上运行 PyTorch 模型的开发者可以据此了解接入方法。
来源:Hugging Face Blog · huggingface.co