Google($GOOGL)公開與Meta($META)的PyTorch團隊共同開發的開源專案TorchTPU,讓TPU成為PyTorch原生支援的硬體選項,與GPU、CPU並列。
Google Cloud官方部落格表示,Google在「PyTorch Conference 2026」上將TorchTPU開源,程式碼已公開於GitHub。現有PyTorch程式碼只需少量修改,就能移植至TPU。
Google Cloud也指出,vLLM、SGLang、TorchTitan、VeRL及Ray等框架已與TorchTPU整合。在Hugging Face Transformers v5中,設定device_map="tpu"即可將模型部署至TPU。
留言 0