LLM推論向けのGPUカーネルライブラリ兼ジェネレーター。attention、GEMM、MoE、サンプリング操作に統一APIを提供します。
awslabs/awsome-distributed-ai
AWSクラスターで分散AIの学習と推論を実行するためのリファレンスアーキテクチャ、テンプレート、サンプル。
- スター
- 495
- フォーク
- 215
- 未解決の Issue
- 38
- 最終コミット
- 16 時間前
これは何?
このリポジトリには、AWSで分散AIの学習と推論を実行するためのリファレンスアーキテクチャ、デプロイ用テンプレート、実行可能なサンプルがまとめられています。Amazon SageMaker HyperPod、AWS ParallelCluster、AWS Parallel Computing Service、Amazon EKSなどのクラスター基盤を対象としており、それぞれにCloudFormationとTerraformのテンプレートが用意されています。
PyTorch DDPやFSDP、Megatron-LM、NeMoといった学習フレームワークのほか、vLLM、SGLang、NVIDIA Dynamoなどのサービングエンジンもサンプルで扱っています。また、カスタムマシンイメージを構築するPackerとAnsibleのスクリプト、クラスターの健全性検証ツール、EFAとNsightの可観測性ガイド、NCCL、NCCOM、NVSHMEMのマイクロベンチマークも含まれています。
AWSでマルチノードGPUの学習や推論を実行するチームは、クラスターのセットアップ、起動スクリプト、パフォーマンスチューニングの出発点として活用できます。
主な機能
- 01SageMaker HyperPod、ParallelCluster、PCS、EKS向けのリファレンスアーキテクチャ
- 02PyTorch FSDP、Megatron-LM、NeMoを使った学習サンプル
- 03vLLM、SGLang、NVIDIA Dynamoを使った推論サンプル
- 04カスタムAMIを作成するPackerとAnsibleのスクリプト
- 05GPUクラスターの健全性チェックとEFAモニタリング用エクスポーター
- 06NCCL、NCCOM、NVSHMEMのマイクロベンチマーク
ユースケース
- 大規模モデルの学習用に、AWS上でSlurmまたはKubernetesのGPUクラスターを構築する
- EKS上でvLLMまたはSGLangを使ったマルチノードLLM推論をデプロイする
- 長時間の学習ジョブを実行する前に、GPUとネットワークの性能を検証する
- NsightとEFAのメトリクスを使って分散ワークロードをプロファイリングする
クイックスタート
- 1
examples ディレクトリを確認し、参考にするアーキテクチャまたはサンプルを選びます
- 2
選んだアーキテクチャまたはサンプルの README を読み、セットアップ手順を確認します
- 3
リポジトリをローカルマシンにクローンします
git clone https://github.com/awslabs/awsome-distributed-ai.git
使う前に知っておきたいこと
- AWS 固有のクラスター(SageMaker HyperPod、ParallelCluster、PCS、EKS)を対象としているため、AWS アカウントとクラスターのセットアップがないと役立ちません。
- 単一のインストール可能なパッケージではなく、参考アーキテクチャとサンプルのコレクションであるため、ワンステップでのインストールはできません。
収録されている特集
トピック
よくある質問
- awsome-distributed-ai とは?
- AWSクラスターで分散AIの学習と推論を実行するためのリファレンスアーキテクチャ、テンプレート、サンプル。 このリポジトリには、AWSで分散AIの学習と推論を実行するためのリファレンスアーキテクチャ、デプロイ用テンプレート、実行可能なサンプルがまとめられています。Amazon SageMaker HyperPod、AWS ParallelCluster、AWS Parallel Computing Service、Amazon EKSなどのクラスター基盤を対象としており、それぞれにCloudFormationとTerraformのテンプレートが用意されています。
- awsome-distributed-ai は無料で使えますか?ライセンスは?
- awsome-distributed-ai は MIT-0 ライセンスのオープンソースです。商用製品で使う前にライセンス条件を確認してください。
- awsome-distributed-ai は今もメンテナンスされていますか?
- 活発にメンテナンス中。awsome-distributed-ai の最終コミットは 2026年11月6日 です。
- awsome-distributed-ai は何に使えますか?
- 大規模モデルの学習用に、AWS上でSlurmまたはKubernetesのGPUクラスターを構築する EKS上でvLLMまたはSGLangを使ったマルチノードLLM推論をデプロイする 長時間の学習ジョブを実行する前に、GPUとネットワークの性能を検証する NsightとEFAのメトリクスを使って分散ワークロードをプロファイリングする
- awsome-distributed-ai のインストール方法は?
- examples ディレクトリを確認し、参考にするアーキテクチャまたはサンプルを選びます 選んだアーキテクチャまたはサンプルの README を読み、セットアップ手順を確認します リポジトリをローカルマシンにクローンします: git clone https://github.com/awslabs/awsome-distributed-ai.git
- awsome-distributed-ai はどこで入手できますか?
- ソースコードとインストール手順は GitHub にあります:https://github.com/awslabs/awsome-distributed-ai
概要はプロジェクトの README をもとに AI が生成しています。GitHub の数値は毎日更新されます。
関連プロジェクト
PyTorch、JAX、XGBoostなどのフレームワークで分散AIモデルの学習やLLMのファインチューニングを行う、Kubernetesネイティブのプラットフォーム。
NCCLは、単一ノードおよびマルチノードの学習向けに、all-reduceやall-gatherなどの最適化されたGPU通信プリミティブを提供するNVIDIAのライブラリです。
大規模言語モデルやマルチモーダルモデルの学習、ファインチューニング、推論実行に役立つ手法、スクリプト、手順をまとめたオープンな書籍。
vLLMを基盤とし、OpenAI互換のサービングと分散デプロイに対応する、Hugging Face互換モデル向けLLM推論エンジンです。