PyTorch 是一個開源的 Python 機器學習庫,基於 Torch 庫,底層由 C++ 實現,應用於人工智慧領域,如電腦視覺和自然語言處理等。

PyTorch 2.4 引入了多項新功能和改進,包括支援 Python 3.12、AOTInductor 凍結功能、新的高階 Python 自訂運算子 API,以及基於 libuv 的 TCPStore 預設伺服器後端,這些提升使得在大規模作業中的效能顯著增強。除此之外,還提供了針對 AWS Graviton 處理器的優化及 BF16 符號形狀優化,為開發者帶來更高效的機器學習框架體驗。

詳細資訊請參考 PyTorch 部落格: