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