iThomeAWS推出PyTorch擴充套件TorchNeuron，降低AI模型轉用Trainium晶片門檻
作者 李建興全文約 907 字
文章摘要
iThome · 文章頁摘要PyTorch擴充套件TorchNeuron讓開發者能在Trainium晶片上繼續使用PyTorch既有的測試、除錯及多晶片訓練工具，並支援逐項執行與編譯執行兩種模式
AWS釋出AI開發框架PyTorch的擴充套件TorchNeuron，讓開發者更容易改用AWS自家AI加速晶片Trainium執行既有模型。開發者在使用受支援的模型與功能時，通常只須調整程式指定的運算裝置，就能維持原有的訓練與推論流程，減少更換晶片時的修改工作。...