vllm-project/aibrix
Cost-efficient and pluggable Infrastructure components for GenAI inference
AIBrix – クイック概要
AIBrix は、大規模言語モデル(LLM)推論を大規模に実行するための オープンソースのクラウドネイティブプラットフォーム です。Kubernetes 上で動作するように構築されており、GenAI サービスのライフサイクル全体(モデルダウンロード、ルーティング、オートスケーリング、障害検出、コストを意識したヘテロジニアス GPU スケジューリング)を処理するコンポーネント群を提供します。
機能
- LLM のデプロイ(LoRA 適用済みバリアントを含む)を Kubernetes リソースとして行います。
- リクエストのルーティング を LLM 対応ゲートウェイを介して適切なモデルレプリカに行います。
- 推論ポッドのオートスケーリング をリアルタイムのトラフィックとサービスレベル目標に基づいて行います。
- KV キャッシュの共有 をノード間で行い、長いコンテキストのプロンプトに対する再計算を削減します。
- GPU 障害の検出 を行い、障害のあるハードウェアをグレースフルに退役させることができます。
- 混合 GPU クラスターのサポート により、低負荷のトラフィックには安価な GPU を、レイテンシが重要なリクエストには高性能 GPU を使用できます。
主要コンポーネント
| コンポーネント | 役割 |
|---|---|
| 統合 AI ランタイム(サイドカー) | メトリクスを標準化し、モデルダウンロードを処理し、軽量な管理 API を提供します。 |
| LLM ゲートウェイとルーティング | クライアント呼び出しのエントリポイントであり、モデルレプリカ間でトラフィックを分散します。 |
| アプリ向けオートスケーラー | Kubernetes HPA を LLM 固有のシグナル(トークンスループット、キューの深さなど)で拡張します。 |
| 分散 KV キャッシュ | ポッド間でアテンションキャッシュを共有し、長いコンテキストの高密度サービスを可能にします。 |
| CRD とオペレーター | オペレーターが調整する宣言的な Kubernetes オブジェクト(例:LLMModel、LLMDeployment)。 |
| GPU 障害検出器 | ハードウェアの健全性を監視し、障害時にポッドの再スケジューリングをトリガーします。 |
想定ユーザー
- エンタープライズ:コストを管理しながら多くの LLM を本番グレードで Kubernetes ネイティブに提供する必要がある企業。
- 研究ラボ:システムレベルの実験用に再現可能でスケーラブルな推論スタックを求める研究機関。
- クラウドプロバイダーまたはプラットフォームチーム:内部ユーザーにマネージド GenAI サービスを提供したいチーム。
はじめに(非常に短い)
# リポジトリをクローン
git clone https://github.com/vllm-project/aibrix.git && cd aibrix
# CRD と依存関係をインストール(ナイトリービルドの例)
kubectl apply -k config/dependency --server-side
kubectl apply -k config/crd --server-side
kubectl apply -k config/default
または、安定版(例:v0.8.0‑rc.1)のリリース済み YAML を使用します。
ドキュメントとコミュニティ
- 完全なドキュメント:https://aibrix.readthedocs.io/latest/
- ブログ、ホワイトペーパー、リリースノートは README からリンクされています。
- コミュニティ Slack:
#aibrix(vLLM 開発ワークスペース経由)。
ライセンス
Apache 2.0 – 商用および学術利用は無料です。
結論: AIBrix は、Kubernetes 上で LLM 推論を実行するために必要な配管をパッケージ化し、モデル管理、ルーティング、オートスケーリング、キャッシュ共有、ハードウェア信頼性をすべて Apache 2.0 ライセンスの下で処理します。
関連
- プロジェクト
- プロジェクト
- プロジェクト
- プロジェクト
- プロジェクト