跳到正文
Hugging Face Blog·· 20 小时前精选AI 评分69

Hugging Face 推出原生 PyTorch 后端 TorchTPU

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

Hugging Face 推出 TorchTPU,为 PyTorch 添加了原生 TPU 后端支持,使 Transformers 和 Diffusers 等库可以直接在 TPU 上运行。

推荐理由

原文介绍了 TorchTPU 如何让 PyTorch 生态原生支持 TPU,并给出了具体的代码示例和性能数据,读者可以据此评估其对工作流的影响。

来源:Hugging Face Blog · huggingface.co