Hugging Face ❤️ TorchTPU:原生支持TPU | AI Archive

Hugging Face 推出了 TorchTPU,这是一个原生的 PyTorch 后端,允许开发者直接在 Google 的张量处理单元(TPU)上运行模型,而无需使用 JAX 或复杂的变通方案。此次更新通过将 TPU 视为普通设备来简化工作流程,使其能够与 Transformers 和 Diffusers 等流行库无缝集成。用户现在可以使用简单的“device_map='tpu'”参数从 Hugging Face Hub 加载模型,并

AI Archive ·

Terminal content

Latest news

News articles

Repository of the day

php/php-src · ⭐ 41,023

The PHP Interpreter

Source: GitHub Trending