蚂蚁集团近日在人工智能领域取得重要突破,正式对外开源其自主研发的高性能强化学习框架Awex(Asystem Weight Exchange framework)。该框架专门针对万亿参数级别模型权重同步需求设计,能够在大规模GPU集群中实现高效并行运算。
基于Awex框架构建的思考型大模型Ring-1T,已在数千张GPU组成的集群环境中完成验证。测试数据表明,该模型能在5至10秒内完成万亿级参数的权重同步任务,显著提升了超大规模模型训练效率。这一技术突破为复杂AI系统的实时协同运算提供了创新解决方案。
作为配套技术生态的重要组成部分,蚂蚁百灵团队上月同步开源了非思考型Ling-1T与思考型Ring-1T两款万亿参数模型。其中Ring-1T通过集成Awex框架的核心功能,在保持模型性能的同时,将参数同步效率提升至行业领先水平。这两款模型的开源将为全球开发者提供更完善的AI基础设施支持。
Awex框架采用分布式权重交换机制,通过优化数据传输路径和计算资源调度,有效解决了大规模模型训练中的通信瓶颈问题。该框架的开源将推动强化学习技术在金融、医疗、智能制造等领域的深度应用,为构建更智能的AI系统奠定技术基础。
