AWS 推出 TorchNeuron:為 Trainium 原生支援 PyTorch(公開測試版)
AWS 宣佈推出 TorchNeuron:為 AWS Trainium 提供原生 PyTorch 支援(目前處於公開測試版階段)
重點導讀
TorchNeuron 是 AWS 為 Trainium AI 晶片推出的原生 PyTorch 後端,已開放公開測試版。
- 功能:使用
.to('neuron')直接在 Trainium 上執行 PyTorch 程式,保持與標準 PyTorch API 完全相容。
推薦理由
TorchNeuron 讓 PyTorch 使用者可在 Trainium 上無縫執行,降低硬體門檻並提升訓練效能。
信源出處aws.amazon.com
AWS 宣佈推出 TorchNeuron:為 AWS Trainium 提供原生 PyTorch 支援(目前處於公開測試版階段)
本文為 888news 精選之重點導讀。若需閱讀作者完整原文、技術細節與官方討論,請前往原始網站。
來源: AWS What's New (aws.amazon.com)
開啟原文