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

Hugging Face发布原生PyTorch后端TorchTPU

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

Hugging Face发布TorchTPU,将TPU作为PyTorch原生后端,并支持Transformers、Diffusers和Kernels。用户可在TPU虚拟机上安装后,通过device_map="tpu"等配置运行Hub模型及图像、视频流水线;项目仍处于测试推进阶段,已运行的TPU测试通过率为96.5%。

推荐理由

TorchTPU将TPU纳入PyTorch原生后端,并贯通Transformers、Diffusers与Kernels;安装方式、设备映射及测试数据可直接用于迁移评估。

来源:Hugging Face Blog · huggingface.co