Amazon 推出開源 TorchNeuron,用於在自研 Trainium 芯片上運行 PyTorch 代碼

icon币界网
分享
AI summary icon精華摘要
Amazon 宣布了關於 TorchNeuron 的鏈上新聞,這是一個開源工具,可讓 PyTorch 代碼在 Trainium 芯片上以最少的設置運行。該工具於 10 月 20 日在聖何塞舉行的 PyTorch 大會上推出,讓開發者無需重寫代碼即可切換芯片。Amazon 表示,這項技術提升了 AI 開發效率並降低了成本。此舉可能影響 AI 和區塊鏈領域的代幣發行新聞。
幣界網報導:

IT之家 10 月 20 日消息,當地時間 10 月 19 日,亞馬遜宣布推出開源工具 TorchNeuron,透過該工具,開發人員編寫的 PyTorch 代碼可直接在亞馬遜自研的 Trainium 芯片上運行,僅需更改一次設定,無需重寫。

IT之家獲悉,TorchNeuron 將於當地時間 10 月 20 日在加利福尼亞州聖何塞舉行的北美 PyTorch 大會上以開源許可形式正式發布。

PyTorch 是由 Meta 主導的開源框架,被大多數人工智慧開發者所採用。英偉達投入了十多年時間研發 CUDA 工具,能夠讓 PyTorch 代碼在英偉達的晶片上高效運行。

據介紹,亞馬遜工程師分別在傳統晶片和 Trainium 晶片上執行了相同的 AI 訓練腳本。唯一的差異僅在於程式碼中指定所用晶片的設定參數。結果顯示,AI 模型在兩種環境下學習到的內容完全一致,學習速度也相同。這意味著開發者原本用於測試、調試和擴展 AI 模型的工具,無需任何修改,即可直接在 Trainium 上使用。

亞馬遜表示,TorchNeuron 為人工智慧行業帶來了一項長期缺失的關鍵要素:真正可行的選擇權。當開發者能夠在不同晶片之間自由切換而無需付出額外代價時,晶片廠商就必須透過提供更具競爭力的產品來贏得開發者的青睞;而開發者則能從中受益,實現成本的降低與迭代速度的提升。

此外,亞馬遜正積極參與 PyTorch 上游開發,與 PyTorch 社區合作,移除將該框架與特定晶片綁定的程式碼。其目標是讓 PyTorch 對整個開發者社區而言更具「硬體無關性」,而不僅僅是服務於 Trainium 用戶。

免責聲明:本頁面資訊可能來自第三方,不一定反映KuCoin的觀點或意見。本內容僅供一般參考之用,不構成任何形式的陳述或保證,也不應被解釋為財務或投資建議。 KuCoin 對任何錯誤或遺漏,或因使用該資訊而導致的任何結果不承擔任何責任。 虛擬資產投資可能存在風險。請您根據自身的財務狀況仔細評估產品的風險以及您的風險承受能力。如需了解更多信息,請參閱我們的使用條款和風險披露 。