近日，天數(shù)智芯通用GPU產(chǎn)品“天垓100”與計(jì)圖即時(shí)編譯深度學(xué)習(xí)框架完成兼容性適配認(rèn)證，同時(shí)支持加速深度學(xué)習(xí)模型的訓(xùn)練和推理，支持運(yùn)行多種前沿AI應(yīng)用，進(jìn)一步助力國產(chǎn)AI生態(tài)發(fā)展。

天垓100 是天數(shù)智芯首款全自研通用GPU產(chǎn)品，支持多精度數(shù)據(jù)類型混合訓(xùn)練，通過豐富全面的自研指令集，釋放強(qiáng)大的可編程性與應(yīng)用通用性，可以提供業(yè)界領(lǐng)先的AI算力密度與能效比。

計(jì)圖(Jittor)是由我國高校研制的開源深度學(xué)習(xí)框架。Jittor框架創(chuàng)新地使用了元算子融合和動(dòng)態(tài)編譯技術(shù)，并提出“統(tǒng)一計(jì)算圖”的思想，使得在多種任務(wù)性能上超越國外主流平臺。

天數(shù)智芯與計(jì)圖團(tuán)隊(duì)、北京非十科技緊密合作，取得了顯著的適配成果。

其一是兼容性強(qiáng)。用戶可以像使用主流產(chǎn)品一樣使用天數(shù)智芯GPU，而無需針對硬件修改代碼，已有的模型可以直接運(yùn)行。

其二是高性能。天垓100的性能得到充分挖掘，Jittor 基于元算子融合和統(tǒng)一計(jì)算圖底層創(chuàng)新，構(gòu)建了高性能的深度學(xué)習(xí)計(jì)算平臺。據(jù)計(jì)圖提供數(shù)據(jù)顯示，在多種骨干網(wǎng)絡(luò)和多個(gè)深度學(xué)習(xí)模型庫上性能顯著超過 PyTorch，平均加速比達(dá)到了4.4倍。

其三是應(yīng)用覆蓋廣。計(jì)圖的骨干網(wǎng)絡(luò)、圖像生成模型庫、遙感解譯庫、圖像分割庫、神經(jīng)輻射場庫等都已經(jīng)可以在天垓100上訓(xùn)練推理，尤其是神經(jīng)輻射場NeRF，被視為目前最為火熱的AI應(yīng)用之一。

關(guān)于天數(shù)智芯

上海天數(shù)智芯半導(dǎo)體有限公司(簡稱“天數(shù)智芯”)于2018年正式啟動(dòng)通用GPU產(chǎn)品設(shè)計(jì)，是中國第一家通用GPU高端芯片及超級算力系統(tǒng)提供商。公司以“成為智能社會的賦能者”為使命，立足客戶、市場的需求，致力于開發(fā)自主可控、國際領(lǐng)先的高性能通用GPU產(chǎn)品，加速AI計(jì)算與圖形渲染融合，探索通用GPU趕超發(fā)展道路，加快建設(shè)自主產(chǎn)業(yè)生態(tài)，打造世界一流的算力引擎，以更可信、更高效、更綠色的算力賦能各行各業(yè)智能化轉(zhuǎn)型，促進(jìn)我國數(shù)字經(jīng)濟(jì)高質(zhì)量發(fā)展，開啟中國引領(lǐng)世界走向元宇宙、數(shù)字孿生的嶄新一頁。

審核編輯：湯梓紅