全球最強AI芯片WSE-3發布 媲美超算 售價超200萬美元?
據了解,美國加州的半導體公司Cerebras Systems公布了其第三代晶圓級AI加速芯片“WSE-3”(Wafer Scale Engine 3),其規格參數令人震驚,而且在保持功耗和價格不變的情況下,性能實現了翻倍,號稱“全球最強”。
據悉,WSE-3采用了臺積電5nm工藝,雖然沒有公布面積,但晶體管數量增加到了夸張的4萬億個,AI核心數量增加到了90萬個,緩存容量達到了44GB,外部搭配內存容量可選1.5TB、12TB、1200TB。雖然核心數量和緩存容量增加不多,但性能卻實現了巨大飛躍,峰值AI算力高達125PFlops,即每秒12.5億億次浮點計算,可媲美頂級超算。
WSE-3可以訓練相當于GPT-4、Gemini十幾倍的下一代AI大模型,能在單一邏輯內存空間內存儲24萬億參數,無需分區或重構。四顆并聯的情況下,一天內即可完成700億參數的調教,支持最多2048路互連,一天便可完成Llama 700億參數的訓練。雖然WSE-3的具體功耗和價格尚未公布,但根據上一代的情況來看,應該會超過200萬美元。
Cerebras Systems公司于2019年推出了第一代WSE-1,基于臺積電16nm工藝,面積為46225平方毫米,晶體管1.2萬億個,擁有40萬個AI核心和18GB SRAM緩存,支持9PB/s內存帶寬和100Pb/s互連帶寬,功耗達15千瓦。隨后在2021年,該公司推出了第二代WSE-2,升級為臺積電7nm工藝,面積保持不變,晶體管數量增加到2.6萬億個,核心數量增加到85萬個,緩存容量增加到40GB,內存帶寬達到20PB/s,互連帶寬達到220Pb/s。