大規模モデルの分散学習と推論を、より簡単かつ効率的にする深層学習最適化ライブラリ。
stas00/ml-engineering
大規模言語モデルやマルチモーダルモデルの学習、ファインチューニング、推論実行に役立つ手法、スクリプト、手順をまとめたオープンな書籍。
- スター
- 19,566
- フォーク
- 1,299
- 未解決の Issue
- 1
- 最終コミット
- 5 日前
これは何?
大規模言語モデルやマルチモーダルモデルの学習とファインチューニング、推論実行のための手法、ツール、手順をまとめたオープンなコレクションです。LLMやVLMの学習エンジニアと運用担当者を対象としており、そのままコピーして使えるコマンドやスクリプトを多数収録しています。
BLOOM-176B、IDEFICS-80B、RAGモデルの学習経験をもとに作られています。ハードウェア(計算資源、ストレージ、ネットワーク)、オーケストレーション(SLURMを含む)、学習、推論、デバッグ、テストを扱うパートに分かれており、一般公開されている学習ログ集も含まれています。
PDFとEPUB形式の電子書籍版に加え、モデルの学習方法やGPUクラスターの評価方法をAIエージェントに教えるためのSKILL.mdファイルを2つ提供しています。
主な機能
- 01大規模学習向けに、アクセラレーター、CPUメモリ、ストレージ、ノード間ネットワークを解説
- 02SLURMのチートシートとオーケストレーションのガイドを収録
- 03ネットワークスループットのベンチマークやノード間接続のテストに使えるツールを提供
- 04ハングまたは失敗するPyTorchアプリケーションのデバッグ手順を紹介
- 05クラウドプロバイダーの選定やGPUアップグレードの判断に役立つガイドを提供
ユースケース
- LLM学習用のマルチノードGPUクラスターを構築・運用する
- ハングまたはクラッシュする分散PyTorch学習ジョブをデバッグする
- GPU世代のアップグレードが費用対効果に見合うかを見積もる
- 新たに納入されたGPUクラスターの受け入れテストを準備する
クイックスタート
- 1
リポジトリをクローンする
git clone https://github.com/stas00/ml-engineering - 2
本の目次を開き、必要なガイドを読む(本はMarkdownファイルで構成されているため、インストールは不要です)
使う前に知っておきたいこと
- インストール可能なソフトウェアではなく、メモやガイドをまとめたものなので、実行するものはありません。
- 内容は主に、著者が特定の大規模トレーニングプロジェクトで得た個人的な経験に基づいています。
トピック
よくある質問
- ml-engineering とは?
- 大規模言語モデルやマルチモーダルモデルの学習、ファインチューニング、推論実行に役立つ手法、スクリプト、手順をまとめたオープンな書籍。 大規模言語モデルやマルチモーダルモデルの学習とファインチューニング、推論実行のための手法、ツール、手順をまとめたオープンなコレクションです。LLMやVLMの学習エンジニアと運用担当者を対象としており、そのままコピーして使えるコマンドやスクリプトを多数収録しています。
- ml-engineering は無料で使えますか?ライセンスは?
- ml-engineering は CC-BY-SA-4.0 ライセンスのオープンソースです。商用製品で使う前にライセンス条件を確認してください。
- ml-engineering は今もメンテナンスされていますか?
- 活発にメンテナンス中。ml-engineering の最終コミットは 2026年10月16日 です。
- ml-engineering は何に使えますか?
- LLM学習用のマルチノードGPUクラスターを構築・運用する ハングまたはクラッシュする分散PyTorch学習ジョブをデバッグする GPU世代のアップグレードが費用対効果に見合うかを見積もる 新たに納入されたGPUクラスターの受け入れテストを準備する
- ml-engineering のインストール方法は?
- リポジトリをクローンする: git clone https://github.com/stas00/ml-engineering 本の目次を開き、必要なガイドを読む(本はMarkdownファイルで構成されているため、インストールは不要です)
- ml-engineering はどこで入手できますか?
- ソースコードとインストール手順は GitHub にあります:https://github.com/stas00/ml-engineering
概要はプロジェクトの README をもとに AI が生成しています。GitHub の数値は毎日更新されます。
関連プロジェクト
大規模AIモデル向けのPyTorchベースの分散学習・推論システムで、並列化、メモリ最適化、異種ハードウェアでの学習を実現します。
学習と推論サービングの両方を対象に、量子化、スパース化、float8学習を提供するPyTorchネイティブのライブラリ。
書籍『Build a Large Language Model (From Scratch)』のサンプルコード。PyTorchでGPT風LLMを段階的に実装します。
Microsoftによる12週間・全24レッスンの初学者向けAIカリキュラム。記号AI、ニューラルネットワーク、コンピュータービジョン、NLP、倫理を扱い、PyTorchとTensorFlowのノートブックを提供します。
ONNX、TensorFlow、PyTorch、Core MLなど、多くの形式に対応したニューラルネットワーク、ディープラーニング、機械学習モデルのビューア。