天垓100率先完成百億級參數大模型訓練，天數智芯迎來新的里程碑
2023-06-12 15:54:36 天數智芯6月10日，在第五屆智源大會AI系統分論壇上，上海天數智芯半導體有限公司（以下簡稱“天數智芯”）對外宣布，在天垓100加速卡的算力集群，基于北京智源人工智能研究院（以下簡稱“智源研究院”）70億參數的Aquila語言基礎模型，使用代碼數據進行繼續訓練，穩定運行19天，模型收斂效果符合預期，證明天數智芯有支持百億級參數大模型訓練的能力。
圖 天數智芯產品線總監宋煜
在北京市海淀區的大力支持下，智源研究院、天數智芯與愛特云翔共同合作，聯手開展基于自主通用GPU的大模型CodeGen（高效編碼）項目，通過中文描述來生成可用的C、Java、Python代碼以實現高效編碼。智源研究院負責算法設計、訓練框架開發、大模型的訓練與調優，天數智芯負責提供天垓100加速卡、構建算力集群及全程技術支持，愛特云翔負責提供算存網基礎硬件及智能化運維服務。
圖 天數智芯大模型訓練全棧方案
在三方的共同努力下，在基于天垓100加速卡的算力集群上，100B Tokens編程語料、70億參數量的AquilaCode大模型參數優化工作結果顯示，1個Epoch后loss下降到0.8，訓練速度達到87K Tokens/s，線性加速比高達95%以上。與國際主流的A100加速卡集群相比，天垓100加速卡集群的收斂效果、訓練速度、線性加速比相當，穩定性更優。在HumanEval基準數據集上，以Pass@1作為評估指標，自主算力集群訓練出來的模型測試結果達到相近參數級別大模型的SOAT水平，在AI編程能力與國際主流GPU產品訓練結果相近。
圖 基于天垓100算力集群的AquilaCode大模型訓練性能采樣
天垓100率先完成百億級參數大模型訓練，邁出了自主通用GPU大模型應用的重要的一步。這一成果充分證明了天垓產品可以支持大模型訓練，打通了國內大模型創新發展的關鍵“堵點”，對于我國大模型自主生態建設、產業鏈安全保障具有十分重大的意義。
接下來，天數智芯將與合作伙伴們繼續深入合作，建設更大規模的天垓100算力集群，完成更大參數規模的大模型訓練，以自主通用GPU產品更好支持國內大模型創新應用，進一步夯實我國算力基礎，助力人工智能產業自主生態建設。
關于智源研究院
北京智源人工智能研究院(簡稱“智源研究院”)是人工智能領域的新型研發機構。2018年11月14日，聯合北京多家人工智能領域優勢單位共建。智源研究院的愿景和目標是聚焦原始創新和核心技術，建立目標導向與自由探索相結合的科研體制，營造全球最佳的學術和技術創新生態，推動北京成為全球人工智能學術思想、基礎理論、頂尖人才、企業創新和發展政策的源頭，率先成為國際領先的人工智能創新中心，推動人工智能產業發展和深度應用，改變人類社會生活，促進人類、環境和智能的可持續發展。
關于天數智芯
上海天數智芯半導體有限公司（簡稱“天數智芯”）是中國領先的通用GPU 高端芯片及超級算力系統提供商。天數智芯以“服務國家戰略”為己任，堅持“四個面向”，致力于開發自主可控、國際領先的高性能通用GPU產品，探索通用GPU趕超發展道路，加快建設自主產業生態，為全產業提供高端算力解決方案。未來，天數智芯將堅持中國特色的自主自強創新發展道路，打造更可信、更高效、更綠色的世界一流算力引擎，賦能千行百業數字化轉型，促進我國數字經濟高質量發展，開啟中國引領世界走向元宇宙、數字孿生的嶄新一頁。
關于愛特云翔
愛特云翔專注為政企客戶提供大數據中心、云計算、人工智能服務，是以“cloud + AI”為技術基座的科技創新型企業。愛特云翔目前擁有淄博臨淄、寧夏中衛、青島即墨等三個大數據產業園和IDC，總機柜數為2.2萬個，已具備覆蓋華北、華中、華東、華南的光纖網絡，建立有北京、上海、淄博軟件與算法研發中心、濟南技術支持中心、深圳硬件設計中心、聊城智能物聯生產線，及全國各地子公司等分支機構。
