亞馬遜云科技發(fā)布新一代AI訓練芯片Amazon Trainium3

12月5日消息，亞馬遜云科技在2024 re:Invent全球大會上，發(fā)布了新一代AI訓練芯片Amazon Trainium3。

據(jù)介紹，Trainium3是亞馬遜云科技首款采用3納米工藝制造的芯片，專為新一代前沿生成式AI工作負載的高性能需求而設(shè)計，在性能、能效和密度上樹立了新標桿。搭載Trainium3的UltraServers性能預計將是Trn2 UltraServers的4倍，使客戶在構(gòu)建模型時能夠更快迭代，并在部署時提供卓越的實時性能。首批基于Trainium3的實例預計將在2025年末上線。

另外，會上還宣布基于Amazon Trainium2的Amazon Elastic Compute Cloud（Amazon EC2）Trn2實例正式可用，并推出全新的Amazon EC2 Trn2 UltraServers，讓用戶以優(yōu)越的性價比訓練和部署最新的人工智能模型，以及未來的大語言模型（LLM）和基礎(chǔ)模型（FM）。

與當前基于GPU的EC2 P5e和P5en實例相比，Amazon EC2 Trn2實例的性價比提升30-40%。它配備16個Trainium2芯片，可提供高達20.8 Petaflops浮點算力的性能。

Amazon EC2 Trn2 UltraServers服務器機型配備64個相互連接的Trainium2芯片，采用NeuronLink超速互連技術(shù)，可提供高達83.2 Petaflops浮點算力，其計算、內(nèi)存和網(wǎng)絡(luò)能力是單一實例的四倍，能夠支持訓練和部署超大規(guī)模的模型。

上周，亞馬遜宣布向Anthropic追加第二筆40億美元投資，并加深在業(yè)務上的合作。

2024 re:Invent全球大會上透露，作為Anthropic與亞馬遜云科技深化合作的一部分，Anthropic已開始優(yōu)化Claude模型，以便在亞馬遜Trainium2上運行。

同時，亞馬遜云科技正在與Anthropic共同構(gòu)建一個名為Project Rainier的 EC2 UltraCluster計算集群，該集群由Trn2 UltraServers組成，能夠在數(shù)十萬個Trainium2芯片之間通過第三代低延遲的PB級EFA網(wǎng)絡(luò)進行分布式模型訓練——其運算能力比Anthropic當前一代領(lǐng)先AI模型訓練所需算力高出5倍以上。該項目完成后，預計將成為全球最大可用的AI計算集群，供Anthropic構(gòu)建和部署未來模型使用。（果青）

生成海報