LoongForge

一套框架,NVIDIA 和昆仑芯通吃,搞定大模型全流程训练

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 模型训练 🍴 复刻 68 🐛 待处理 32 👀 关注 3
访问官网 ↗ GitHub ↗
GitHub 星标 ★ 836
维护状态 活跃
是否开源 开源
定价模式 🏷️完全免费 开源项目,免费使用

项目预览

LoongForge 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类模型训练
开发团队baidu-baige
定价模式🏷️完全免费
价格说明开源项目,免费使用
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议Apache-2.0
主要语言Python
技术栈/模型gpu,kunlun,llm,lora,megatron,mid-training,pretraining,sft,torch,vla,vlm,wam,wan,xpu
GitHub 星标★ 836
上线时间2026-04-23 00:00:00
代码更新2026-11-11
仓库体积93.4 MB
本站数据更新2026-11-11
维护状态活跃
收录时间2026-08-13
浏览次数12

项目介绍

LoongForge 是一个面向大模型训练的统一高性能框架,支持在 NVIDIA GPU 和昆仑芯 XPU 上训练 LLM、VLM、扩散模型及具身智能模型。它解决了多模态与多硬件环境下训练框架割裂、迁移成本高的问题,提供从预训练、中训、SFT 到 LoRA 微调的完整训练流程,并集成 Megatron 等分布式策略。核心能力包括:统一的模型定义与训练接口,屏蔽底层硬件差异;针对昆仑芯 XPU 的深度优化,填补国产芯片训练框架空白;支持扩散模型和 VLA(视觉-语言-动作)等前沿任务,适配 Wan 等视频生成模型。项目处于早期阶段,代码结构清晰,适合研究机构和企业进行国产化训练栈探索。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-13 · 当前星标 ★ 836

技术标签

使用教程

难度:高级 约 90 分钟 部署方式:Docker 部署 4 步

⚠ 该项目 README 未提供完整安装说明,以下步骤为通用流程,请以官方文档为准。

环境要求

  • 多卡 GPU(建议 8 卡)
  • CUDA 12.x
  • 大容量存储

安装与启动步骤

  1. 1准备 GPU 环境

    百度的大模型训练框架,多后端架构。

    nvidia-smi && docker --version
  2. 2拉镜像

    官方提供带 CUDA 的镜像。

    docker pull <loongforge 镜像>
  3. 3起容器

    注意共享内存与 IPC 设置。

    docker run --gpus all --ipc=host -it --rm <image> /bin/bash
  4. 4跑示例训练

    先用最小配置验证环境。

    bash scripts/train_example.sh

如何确认成功

示例训练能跑完若干 step,loss 正常下降。

注意事项

  • 不同模型族与规模要用不同后端,配置里别选错。
  • 分布式训练对网络要求高,单机多卡先跑通。
  • 文档以中文为主,先读 quickstart。

核心亮点

  • 同时支持 NVIDIA GPU 和昆仑芯 XPU,降低国产芯片迁移成本
  • 覆盖预训练、SFT、LoRA、扩散模型和 VLA,训练场景全面
  • 集成 Megatron 分布式能力,适合大规模并行训练

不足之处

  • 早期项目,文档和社区生态待观察
  • 昆仑芯 XPU 的硬件普及度有限,实际验证场景可能较少

适用场景

  • 在国产算力集群上训练 LLM 或 VLM
  • 多模态扩散模型(如视频生成)的分布式训练
  • 具身智能 VLA 模型的微调与预训练

替代项目

Megatron-LM、DeepSpeed、ColossalAI

上一篇:pruna

下一篇:lightly-train

同类项目推荐

shortened-llm 开源

压缩LLM,推理提速,显存减半,生成依旧流畅。

Compressed LLMs for Efficient Text Generation [ICLR'24 Workshop]

★ 91 2026-08-09
VeriThinker 开源

让模型学会自我验证,推理更快更准

[NeurIPS 2025] VeriThinker: Learning to Verify Makes Reasoning Model Efficient

★ 69 2026-08-09
spiral 开源

让AI在对抗博弈中练出深度推理

SPIRAL: Self-Play on Zero-Sum Games Incentivizes Reasoning via Multi-Agent Multi-Tur···

★ 211 2026-08-09
ROLL 开源

大模型强化学习训练,一键从单机扩到集群

An Efficient and User-Friendly Scaling Library for Reinforcement Learning with Large···

★ 3453 2026-09-15
AutoTimes 开源

用大模型直接做时间序列预测,无需改架构

Official implementation for "AutoTimes: Autoregressive Time Series Forecasters via L···

★ 270 2026-08-09
gpu-recipes 开源

在 GCP 上按配方跑通大模型训练与推理基准

Recipes for reproducing training and serving benchmarks for large machine learning m···

★ 149 2026-10-17
MPO 开源

让LLM智能体先学会定计划,再动手执行,提升复杂任务成功率

MPO: Boosting LLM Agents with Meta Plan Optimization (EMNLP 2025 Findings)

★ 82 2026-08-09
doc-to-lora 开源

让大模型随时记住新知识,不用重训

Hypernetworks that update LLMs to remember factual information

★ 843 2026-08-09