注册

亚马逊发布Trainium3 AI芯片

来源:财新网       

在2025年12月2日的AWS re:Invent大会上,亚马逊云端服务(AWS)正式推出其最新一代AI芯片Trainium3。Trainium3是AWS首款采用3纳米制程技术的AI芯片,专为训练与服务下一代生成式AI、推理、多模态与视频生成等应用而设计。

根据AWS官方资料,Trainium3芯片单颗提供高达2.52 petaflops(PFLOPs)的FP8运算效能,内存容量提升1.5倍、频宽提升1.7倍,达到144 GB HBM3e内存与4.9 TB/s内存频宽。Trainium3 UltraServer系统整合最多144颗Trainium3芯片,单一系统可提供最高20.7 TB HBM3e内存、706 TB/s内存频宽与362 PFLOPs FP8运算效能,相较于前一代Trainium2 UltraServer,运算效能提升4.4倍、内存频宽提升3.9倍,能效提升超过4倍。

Trainium3支持先进数据格式(如MXFP8与MXFP4),并针对密集与专家平行(expert-parallel)工作负载进行优化,适合强化学习、Mixture-of-Experts(MoE)、推理与长上下文架构等应用。AWS同时扩展Neuron开发框架,支持原生PyTorch与开源组件,让开发者能更轻松地在Trainium3上部署与优化AI模型。

此外,AWS也宣布Trainium2芯片已全面供不应求,Trainium业务已成长为数十亿美元规模,并在2025年第三季成长150%。Trainium3将扩大服务对象,从原本少数大型客户扩展至更多企业用户。

在模型方面,AWS同步推出Nova 2系列第二代自研模型,包含Lite、Pro、Sonic与Omni四款新模型,进一步提升推理与多模态处理效能,支持更复杂的AI应用场景。