跳到正文
Hugging Face Blog(huggingface.co)·· 18 小時前精選AI 評分82

Hugging Face 推出 TorchTPU:PyTorch 原生支援 TPU

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 導讀

TorchTPU 讓 PyTorch 直接在 TPU 上執行,省去 torch_xla 與 JAX 的中介層。

  • 核心功能:新增 TPU 作為 PyTorch 後端,支援 eager 與 torch.compile,提供 Pallas kernels、分散式訓練。
推薦理由

TorchTPU 讓 PyTorch 直接在 TPU 上執行,省去 torch_xla 與 JAX 的中介層,提升開發效率與推理速度。

信源出處huggingface.co

Hugging Face ❤️ TorchTPU: TPUs, Natively

本文為 888news AI 精選之重點導讀。若需閱讀作者完整原文、技術細節與官方討論,請前往原始網站。

開啟原文
來源: Hugging Face Blog (huggingface.co)
開啟原文