3月14日消息，面對火爆的AI市場，新創(chuàng)AI芯片公司Cerebras Systems近日推出了其第三代的晶圓級AI芯片WSE-3，性能達(dá)到了上一代WSE-2的兩倍，將用于訓(xùn)練業(yè)內(nèi)一些最大的人工智能模型。

據(jù)介紹，WSE-3依然是采用了一整張12英寸晶圓來制作，基于臺積電5nm制程，芯片面積為46225平方毫米，擁有的晶體管數(shù)量達(dá)到了4萬億個(gè)，擁有90萬個(gè)AI核心，44GB片上SRAM，整體的內(nèi)存帶寬為21PB/s，結(jié)構(gòu)帶寬高達(dá)214PB/s。使得WSE-3具有125 FP16 PetaFLOPS的峰值性能，相比上一代的WSE-2提升了1倍。

作為對比，WSE-2芯片面積同樣是46225平方毫米，基于臺積電7nm制程，晶體管數(shù)量為2.6萬億個(gè)，AI內(nèi)核數(shù)量為85萬個(gè)，片上內(nèi)存SRAM為40GB，內(nèi)存帶寬為20PB/s，結(jié)構(gòu)帶寬高達(dá)220PB/s。

如果將其與英偉達(dá)的H100相比，WSE-3面積將是H100的57倍，內(nèi)核數(shù)量是H100的52倍，片上內(nèi)存是H100的880倍，內(nèi)存帶寬是H100的7000倍，結(jié)構(gòu)帶寬是H100的3715倍。

與此同時(shí)，Cerebras還推出了基于WSE-3的CS-3超級計(jì)算機(jī)，可用于訓(xùn)練參數(shù)高達(dá)24萬億的人工智能模型，這比相比基于WSE-2和其他現(xiàn)代人工智能處理器的超級計(jì)算機(jī)有了重大飛躍。該超級計(jì)算機(jī)可以支持1.5TB、12TB或1.2PB的外部內(nèi)存，這使它能夠在單個(gè)邏輯空間中存儲大量模型，而無需分區(qū)或重構(gòu)，從而簡化了訓(xùn)練過程，提高了開發(fā)人員的效率。

最新的 Cerebras 軟件框架可以為PyTorch 2.0 和最新的 AI 模型和技術(shù)（如多模態(tài)模型、視覺轉(zhuǎn)換器、專家混合和擴(kuò)散）提供原生支持。Cerebras 仍然是唯一為動態(tài)和非結(jié)構(gòu)化稀疏性提供本機(jī)硬件加速的平臺，相比英偉達(dá)的DGX-100計(jì)算機(jī)系統(tǒng)，將訓(xùn)練速度提高了8 倍。

Cerebras還強(qiáng)調(diào)了CS-3卓越的電源效率和易用性。盡管性能翻了一番，CS-3仍保持著與前代產(chǎn)品相同的功耗。它還簡化了大型語言模型（LLM）的訓(xùn)練，與GPU相比，所需代碼減少了97%。例如，該公司表示，GPT-3大小的模型在Cerebras平臺上只需要565行代碼。

就可擴(kuò)展性而言，CS-3可以配置在多達(dá)2048個(gè)系統(tǒng)的集群中。這種可擴(kuò)展性使其能夠在四個(gè)系統(tǒng)的設(shè)置下，在一天內(nèi)微調(diào)700億個(gè)參數(shù)模型，并在同一時(shí)間段內(nèi)全面從頭開始訓(xùn)練Llama 70B模型。

Cerebras表示，該公司已經(jīng)看到了外界對CS-3的巨大興趣，并有來自各個(gè)行業(yè)的大量訂單積壓，包括企業(yè)、政府和公有云廠商。Cerebras還與美國阿貢國家實(shí)驗(yàn)室和梅奧診所等機(jī)構(gòu)合作，強(qiáng)調(diào)了CS-3在醫(yī)療保健方面的潛力。

Cerebras和G42之間的戰(zhàn)略合作伙伴關(guān)系也將隨著Condor Galaxy 3的建設(shè)而擴(kuò)大，這是一款人工智能超級計(jì)算機(jī)，擁有64個(gè)CS-3系統(tǒng)（擁有57600000個(gè)核心）。這兩家公司已經(jīng)共同創(chuàng)建了世界上最大的兩臺人工智能超級計(jì)算機(jī)：Condor Galaxy 1 （CG-1）和Condor Galaxy 2（CG-2），它們位于加利福尼亞州，綜合性能為8個(gè)ExaFLOP。這一合作伙伴關(guān)系的目標(biāo)是在全球范圍內(nèi)提供數(shù)十個(gè)exaFLOP的人工智能計(jì)算。

G42集團(tuán)首席技術(shù)官Kiril Evtimov表示：“我們與Cerebras的戰(zhàn)略合作伙伴關(guān)系有助于推動G42的創(chuàng)新，并將有助于加速全球范圍內(nèi)的人工智能革命?！薄！癈ondor Galaxy 3是我們的下一臺人工智能超級計(jì)算機(jī)，擁有8個(gè)exaFLOP，目前正在建設(shè)中，很快將使我們系統(tǒng)的人工智能計(jì)算總產(chǎn)量達(dá)到16個(gè)exafloP。”

相關(guān)文章：《Cerebras推出全球最強(qiáng)AI超算：5400萬個(gè)AI內(nèi)核，算力高達(dá)4 exaFLOPS！》

編輯：芯智訊-浪客劍