高效可扩展训练部署子系统LibAI模型库代码
收藏国家基础学科公共科学数据中心2026-01-30 收录
下载链接:
https://nbsdc.cn/general/dataDetail?id=686a8dcb195d2621a90dbfd3&type=1
下载链接
链接失效反馈官方服务:
资源简介:
LibAI模型库代码(Libai Code)主要面向大规模预训练模型的研究与应用需求建设,由Oneflow-Inc团队基于OneFlow深度学习框架开发并在GitHub平台开源发布,源码地址为:https://github.com/Oneflow-Inc/libai。该库的开发背景是为适应当前自然语言处理、计算机视觉和多模态任务对高效模型实现、高性能训练及推理框架的迫切需求,特别强调在国产化算力平台上的工程可用性和性能优化能力。LibAI整合了高性能的模型组件库、自动并行机制、通用训练流程与评估模块,支持如BERT、GPT、ViT、Swin Transformer等主流模型结构,具备较强的可扩展性和工程部署适配能力。项目代码体量超过数万行,涵盖配置管理、数据加载、训练调度、模型构建、评估输出等多个核心模块,支持在GPU和国产NPU上高效运行。LibAI是科研实验与工程实践结合的重要资源,便于研究人员快速验证新算法、模型结构和分布式训练策略。
提供机构:
北京一流科技有限公司



