幣界網報導:
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 用戶。


