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