Hugging Face Blog·· 11 天前精选
Hugging Face ❤️ TorchTPU：TPU 原生支持
Hugging Face ❤️ TorchTPU: TPUs, Natively
原始来源：Hugging Face Blog阅读原始报道
AI 导读
Hugging Face 推出 TorchTPU，使 TPU 成为 PyTorch 的原生后端。用户可通过 device_map='tpu' 直接在 TPU 上加载并运行 Transformers 和 Diffusers 模型。
推荐理由
TorchTPU 让 TPU 成为 PyTorch 的原生后端，用户只需一行 device_map='tpu' 即可在 TPU 上运行 Transformers、Diffusers 等库，并提供分布式训练和编译加速等功能。
来源：Hugging Face Blog · huggingface.co