Hugging Face on PyTorch / XLA TPUs

Hugging Face 宣布支援 PyTorch/XLA，讓開發者能在 Google Cloud TPU 上輕鬆訓練與部署 Transformer 模型。

Hugging Face 於 2021 年宣布與 Google 合作，正式支援 PyTorch/XLA。這項整合讓 PyTorch 開發者無需修改大量程式碼，即可利用 Google Cloud TPU 的強大算力來加速 Transformer 模型的訓練與推理。透過 Hugging Face 的 Trainer API，開發者可以輕鬆在單個或多個 TPU 節點上進行分散式訓練，大幅提升運算效率並降低成本。

摘要由 AI 整理,以原文為準。