AXLearn
A
Axlearn
簡介 :
AXLearn是Apple基於JAX和XLA構建的深度學習庫,採用面向對象的方式解決軟件工程在大規模深度學習模型開發中的挑戰。它的配置系統允許用戶從可重用的構建塊中組合模型,並與其他庫(如Flax和Hugging Face transformers)集成。AXLearn旨在擴展訓練規模,支持數百億參數的模型在數千個加速器上高效訓練,適合在公有云上部署。它還採用全局計算範式,允許用戶描述全局虛擬計算機上的計算,而不是單個加速器。AXLearn支持廣泛的應用,包括自然語言處理、計算機視覺和語音識別,幷包含用於訓練最先進模型的基線配置。
需求人群 :
用於開發和訓練大規模深度學習模型
總訪問量: 474.6M
佔比最多地區: US(18.64%)
本站瀏覽量 : 51.6K
使用場景
使用AXLearn的Transformer模型訓練語言模型
利用AXLearn在數百個GPU上訓練圖像分類模型
構建語音識別pipeline,通過AXLearn的配置系統集成各個組件
產品特色
面向對象的軟件工程方法
可擴展的模型訓練
全局計算範式
集成其他深度學習庫
公有云部署支持
AIbase
智啟未來,您的人工智能解決方案智庫
© 2025AIbase