跳到正文
AWS What's New(aws.amazon.com)· aws@amazon.com·· 11 天前精選品質評分68

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)
開啟原文