大規模モデルの分散学習と推論を、より簡単かつ効率的にする深層学習最適化ライブラリ。
ServiceNow/Fast-LLM
PyTorchとTritonを基盤に、3D並列処理、ZeROによるメモリ最適化、Hugging Face連携を備えた大規模言語モデルの学習用オープンソースライブラリ。
- スター
- 337
- フォーク
- 53
- 未解決の Issue
- 56
- 最終コミット
- 6 日前
これは何?
Fast-LLMは、PyTorchとTritonを基盤とする大規模言語モデル学習用のオープンソースライブラリです。約10億パラメータのモデルから、700億超のパラメータを持つモデルを学習するクラスターまでを対象とし、単一GPUから、SlurmまたはKubeFlowを使ったKubernetesで管理されるマルチノードクラスターまで対応します。
データ並列、テンソル並列、パイプライン並列、シーケンス長並列による分散学習に加え、メモリ効率を高めるZeRO-1、ZeRO-2、ZeRO-3をサポートしています。モデルはYAMLファイルで設定し、コマンドラインインターフェースから起動できます。また、トークンをドロップしないMixture-of-Experts実装を備え、Hugging Face Transformersと連携します。
ライセンスはApache 2.0で、すばやくデプロイできるビルド済みDockerイメージも公開されています。ドキュメントとサンプルでは、READMEで使用されているMistral-7Bの学習設定を紹介しています。
主な機能
- 01データ、テンソル、パイプラインによる3D並列処理を使った分散学習。
- 02メモリ最適化のためのZeRO-1、ZeRO-2、ZeRO-3。
- 03トークンをドロップしないMixture-of-Experts(MoE)実装。
- 04混合精度学習と勾配累積。
- 05YAMLベースの設定とコマンドラインインターフェース。
- 06Hugging Face Transformersとの連携。
ユースケース
- Slurmで管理されるマルチノードGPUクラスターで、Mistral系またはGPT系モデルを事前学習する。
- KubeFlow PyTorchJobを使って、Kubernetes上で大規模なLLM学習ジョブを実行する。
- Apache 2.0ライセンスのオープンソースコードベースを使って、カスタムMoEモデルを学習する。
クイックスタート
- 1
nvcr.io/nvidia/pytorch:25.11-py3 Docker イメージをすべてのノードにデプロイします(推奨)
- 2
pip を使って Fast-LLM をすべてのノードにインストールします
pip install --no-cache-dir "fast-llm[CORE,OPTIONAL] @ git+https://github.com/ServiceNow/Fast-LLM.git" - 3
サンプルの Slurm ジョブスクリプトをクラスターに送信します
sbatch examples/fast-llm.sbat
使う前に知っておきたいこと
- 掲載されている規模でトレーニングするには、各ノードに A100 または H100 GPU を 8 基搭載した 4 ノード構成などのマルチノード GPU クラスターが必要です。
- すべてのノードに CUDA 12.1 以降と、PyTorch、Triton、Apex がインストールされている必要があります。
- README によると、カスタマイズ可能なオプティマイザーやチュートリアルなど、一部の機能はまだ開発中です。
トピック
よくある質問
- Fast-LLM とは?
- PyTorchとTritonを基盤に、3D並列処理、ZeROによるメモリ最適化、Hugging Face連携を備えた大規模言語モデルの学習用オープンソースライブラリ。 Fast-LLMは、PyTorchとTritonを基盤とする大規模言語モデル学習用のオープンソースライブラリです。約10億パラメータのモデルから、700億超のパラメータを持つモデルを学習するクラスターまでを対象とし、単一GPUから、SlurmまたはKubeFlowを使ったKubernetesで管理されるマルチノードクラスターまで対応します。
- Fast-LLM は無料で使えますか?ライセンスは?
- Fast-LLM は Other ライセンスのオープンソースです。商用製品で使う前にライセンス条件を確認してください。
- Fast-LLM は今もメンテナンスされていますか?
- 活発にメンテナンス中。Fast-LLM の最終コミットは 2026年10月15日 です。
- Fast-LLM は何に使えますか?
- Slurmで管理されるマルチノードGPUクラスターで、Mistral系またはGPT系モデルを事前学習する。 KubeFlow PyTorchJobを使って、Kubernetes上で大規模なLLM学習ジョブを実行する。 Apache 2.0ライセンスのオープンソースコードベースを使って、カスタムMoEモデルを学習する。
- Fast-LLM のインストール方法は?
- nvcr.io/nvidia/pytorch:25.11-py3 Docker イメージをすべてのノードにデプロイします(推奨) pip を使って Fast-LLM をすべてのノードにインストールします: pip install --no-cache-dir "fast-llm[CORE,OPTIONAL] @ git+https://github.com/ServiceNow/Fast-LLM.git" サンプルの Slurm ジョブスクリプトをクラスターに送信します: sbatch examples/fast-llm.sbat
- Fast-LLM はどこで入手できますか?
- ソースコードとインストール手順は GitHub にあります:https://github.com/ServiceNow/Fast-LLM
概要はプロジェクトの README をもとに AI が生成しています。GitHub の数値は毎日更新されます。
関連プロジェクト
大規模AIモデル向けのPyTorchベースの分散学習・推論システムで、並列化、メモリ最適化、異種ハードウェアでの学習を実現します。
コンピュータビジョン向けに、画像モデルのアーキテクチャ、事前学習済み重み、学習・検証・エクスポート用スクリプトを提供するPyTorchライブラリ。
画像分類などのビジョンタスク向けに、Vision Transformer(ViT)と多数の派生モデルを実装したPyTorchライブラリです。
Mambaは、選択的状態空間モデルを基盤とするPyTorchのシーケンスモデリングアーキテクチャで、言語などのシーケンス処理においてTransformerに代わる線形時間の手法を提供します。
GPTのTransformerのように並列学習できる、RNN型言語モデルアーキテクチャRWKVの公式学習・リファレンスコード。