CalvinXKY/InfraTech

分享AI Infra知识&代码练习:PyTorch、vLLM/SGLang、slime/vime框架入门⚡️、性能加速🚀、大模型基础🧠、AI软硬件🔧等

解決する課題

InfraTechは、AIモデルの背後にある複雑なインフラストラクチャを理解するための構造化された学習パスと実践的なコード例を提供します。高レベルのモデル利用と、大規模なAIモデルを効率的にトレーニングおよび実行するために必要な低レベルのシステムエンジニアリングとの間のギャップを埋めます。

仕組み

このプロジェクトは、厳選されたJupyterノートブックと技術ドキュメントのコレクションとして構成されています。分散トレーニング、推論最適化、ハードウェアアクセラレーションなどの主要な概念をカバーしています。ユーザーは、vLLMスケジューリング、SGLangメモリ管理、カスタムPyTorchトレーニングループなどの特定の実装を探索して、これらのシステムが内部でどのように機能するかを確認できます。

対象読者

このリポジトリは、AIインフラストラクチャを専門にしたい開発者、研究者、AIエンジニア向けに設計されています。モデルデプロイメントの仕組み、パフォーマンスチューニング、人気フレームワークの内部アーキテクチャを習得したいと考えている人に最適です。

ハイライト

  • 推論とトレーニングの内部を実践的に学習するための実用的なJupyterノートブック。
  • vLLMやSGLangなどの人気フレームワークへの深い掘り下げ。カスタムスケジューラとメモリ管理の実装を含む。
  • 並列戦略(TP/PP/SP/EP)、量子化、投機的復号化などのパフォーマンス最適化技術の包括的なカバレッジ。
  • 集合通信や強化学習(RL)のトレーニング-推論併置など、分散トレーニングの基礎の説明。
  • ハードウェアを意識したAIエンジニアリングに関する厳選された技術記事とドキュメント。

関連

  • プロジェクト
  • プロジェクト
  • プロジェクト
  • プロジェクト