跳到正文
Hugging Face Blog·· 13 天前AI 评分22

Hugging Face 拥抱 TorchTPU:PyTorch 原生支持 TPU

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

Hugging Face 宣布 TorchTPU 为 PyTorch 增加 TPU 原生后端,Transformers、Diffusers 和 Kernels 可直接在 TPU 上运行,模型通过 device_map="tpu" 或 pipe.to("tpu") 即可加载。

来源:Hugging Face Blog · huggingface.co