量子化なしで、レイヤー単位に大規模言語モデルを実行し、70Bクラスのモデルを4GB GPU 1枚で推論できるPythonライブラリ。
baidu-baige/LoongForge
Megatron-LMとtorch-nativeバックエンドを基盤に、NVIDIA GPUとKunlun XPUでLLM、VLM、拡散モデル、身体性モデルを学習するためのフレームワーク。
- スター
- 694
- フォーク
- 58
- 未解決の Issue
- 30
- 最終コミット
- 18 時間前
これは何?
LoongForgeは、Baidu AI Cloud Baigeチームが開発した学習フレームワークです。LLM、視覚言語モデル、拡散モデル、VLAやworld-actionモデルなどの身体性モデルを対象に、事前学習、中間学習、SFT、LoRAに対応します。LLM、VLM、拡散モデルのワークロードにはMegatronバックエンドを使い、身体性モデルにはDDP、ZeRO-1、FSDP、HSDPに対応するtorch-nativeバックエンドを使います。
40種類以上のオープンソースモデル向けに、すぐに実行できる起動スクリプトと設定ファイルを提供しています。Megatron形式とHugging Face形式の間でチェックポイントを変換するツールや、データ前処理ツールも備えています。プラグイン設計によりNVIDIA GPUとKunlun XPUの両方に対応し、MoEのエキスパート並列通信のオーバーラップ、FP8学習、コンテキスト並列化、VLMのコンポーネントごとに異なる並列化方式を設定する機能などを搭載しています。
Python製で、ライセンスはApache-2.0です。Dockerイメージに加え、インストール手順やチュートリアルを掲載したドキュメントも提供しています。
主な機能
- 01LLM、VLM、拡散モデル、身体性モデルの学習に対応するMegatronおよびtorch-nativeバックエンド。
- 0240種類以上のオープンソースモデル向けに、すぐに使える起動スクリプトと設定ファイルを用意。
- 03NVIDIA GPUとKunlun XPUで実行可能。
- 04事前学習、中間学習、SFT、LoRAに対応。
- 05Megatron形式とHugging Face形式のチェックポイント変換、およびデータ前処理ツールを搭載。
- 06MoEの通信最適化、FP8学習、コンテキスト並列化に対応。
ユースケース
- マルチノードGPUクラスターで、大規模なMoE言語モデルを事前学習またはファインチューニングする。
- ViT部分とLLM部分で並列化設定を個別に指定し、視覚言語モデルを学習する。
- LIBEROデータを使い、Pi0.5などのVLAロボットポリシーをファインチューニングする。
- Kunlun XPUハードウェアで大規模モデルの学習ワークロードを実行する。
使う前に知っておきたいこと
- NVIDIA GPUまたはKunlun XPUが必要なため、ほかのハードウェアは対象としていません。
- ソースコードが公開されたのは2026年4月で、最初のタグ付きリリースもv0.1.0のため、まだ新しいプロジェクトです。
トピック
よくある質問
- LoongForge とは?
- Megatron-LMとtorch-nativeバックエンドを基盤に、NVIDIA GPUとKunlun XPUでLLM、VLM、拡散モデル、身体性モデルを学習するためのフレームワーク。 LoongForgeは、Baidu AI Cloud Baigeチームが開発した学習フレームワークです。LLM、視覚言語モデル、拡散モデル、VLAやworld-actionモデルなどの身体性モデルを対象に、事前学習、中間学習、SFT、LoRAに対応します。LLM、VLM、拡散モデルのワークロードにはMegatronバックエンドを使い、身体性モデルにはDDP、ZeRO-1、FSDP、HSDPに対応するtorch-nativeバックエンドを使います。
- LoongForge は無料で使えますか?ライセンスは?
- LoongForge は Apache-2.0 ライセンスのオープンソースです。商用製品で使う前にライセンス条件を確認してください。
- LoongForge は今もメンテナンスされていますか?
- 活発にメンテナンス中。LoongForge の最終コミットは 2026年10月19日 です。
- LoongForge は何に使えますか?
- マルチノードGPUクラスターで、大規模なMoE言語モデルを事前学習またはファインチューニングする。 ViT部分とLLM部分で並列化設定を個別に指定し、視覚言語モデルを学習する。 LIBEROデータを使い、Pi0.5などのVLAロボットポリシーをファインチューニングする。 Kunlun XPUハードウェアで大規模モデルの学習ワークロードを実行する。
- LoongForge はどこで入手できますか?
- ソースコードとインストール手順は GitHub にあります:https://github.com/baidu-baige/LoongForge
概要はプロジェクトの README をもとに AI が生成しています。GitHub の数値は毎日更新されます。
関連プロジェクト
WindowsやLinuxで、一般向けNVIDIA・AMD GPU上に大規模なQwen3.8-Flash-Nextモデルを実行するローカル推論エンジン。OpenAI・Anthropic互換APIに対応。
GPU、CPU、RAMを検出し、パラメータ数ではなくベンチマークに基づいて、実際にハードウェア上で動作するHuggingFaceのローカルLLMを順位付けするCLIです。
Qwen3.5とC-RADIOv4-Hを基盤とする、視覚理解・空間推論・ツール利用・身体性AI研究向けの90億パラメータのオープンウェイトマルチモーダルモデル。
FSDP2、Muon、Liger kernelsに対応した、VLM・diffusion・LLM向けの統合トレーニングエンジン。
12 GBのRTX 4070で、262kトークンのコンテキストウィンドウとq8_0 K/V cacheを使ってMirai S 27B(Qwen3.8-27B GGUF)をサービングする構成です。