事前学習、ファインチューニング、評価、サービングに対応する、20種類以上のオープンウェイトモデルをスクラッチ実装したPyTorchベースのLLMツールキット。
kvcache-ai/ktransformers
限られたGPUメモリで大規模MoEモデルを推論し、LoRAまたは全パラメータをファインチューニングするためのCPU-GPU異種構成LLM研究フレームワーク。
- スター
- 19,621
- フォーク
- 1,602
- 未解決の Issue
- 526
- 最終コミット
- 一昨日
これは何?
KTransformersは、CPUとGPUで処理を分担して大規模言語モデルを実行・ファインチューニングするための研究プロジェクトです。推論機能では、INT4/INT8量子化重みに対応したCPU最適化カーネルを提供し、Mixture-of-Expertsモデル向けにAMXやAVX512/AVX2による高速化とNUMAを考慮したメモリ管理を備えています。使用頻度の高いエキスパートはGPUに配置し、低いエキスパートはCPUに置けます。また、カーネルはSGLangとの連携用にPython APIを提供します。
ファインチューニング機能はLLaMA-Factoryと統合されており、限られたGPUメモリでDeepSeek-V3/R1、Qwen3 MoE、Kimi K2の各種モデルなど、超大規模MoEモデルのLoRAおよび全パラメータ学習に対応します。さらに、BF16、FP8、INT8の各種設定、リソース計画、トラブルシューティングに関するドキュメントを公開しており、異種構成によるMoEファインチューニング手法についての研究論文も公開しています。
主な機能
- 01INT4/INT8量子化推論向けに最適化されたIntel AMXおよびAVX512/AVX2カーネル
- 02MoEモデルのCPU-GPUエキスパートスケジューリング。使用頻度の高いエキスパートはGPUに、低いエキスパートはCPUに配置
- 03SGLang連携用Python API
- 04LLaMA-Factoryと統合したLoRAおよび全パラメータのファインチューニング
- 05BF16、FP8、INT8の学習設定に対応
ユースケース
- 大容量メモリのCPUサーバーと組み合わせた単一GPUで大規模MoEモデルを実行する
- CPU-GPU混在のエキスパート配置で、SGLang経由でMoEモデルをサービングする
- LLaMA-Factoryを使い、少数のコンシューマー向けGPUで超大規模MoEモデルをファインチューニングする
クイックスタート
- 1
kt-kernel のソースディレクトリに移動する
cd kt-kernel - 2
ソースから kt-kernel をインストールする
pip install .
使う前に知っておきたいこと
- 特殊なハードウェア向けに構築されており、ヘテロジニアス推論には Intel AMX または AVX512/AVX2 対応の CPU と GPU が必要です。
- 研究プロジェクトとして位置付けられており、元の統合フレームワークはアーカイブされています。
トピック
よくある質問
- ktransformers とは?
- 限られたGPUメモリで大規模MoEモデルを推論し、LoRAまたは全パラメータをファインチューニングするためのCPU-GPU異種構成LLM研究フレームワーク。 KTransformersは、CPUとGPUで処理を分担して大規模言語モデルを実行・ファインチューニングするための研究プロジェクトです。推論機能では、INT4/INT8量子化重みに対応したCPU最適化カーネルを提供し、Mixture-of-Expertsモデル向けにAMXやAVX512/AVX2による高速化とNUMAを考慮したメモリ管理を備えています。使用頻度の高いエキスパートはGPUに配置し、低いエキスパートはCPUに置けます。また、カーネルはSGLangとの連携用にPython APIを提供します。
- ktransformers は無料で使えますか?ライセンスは?
- ktransformers は Apache-2.0 ライセンスのオープンソースです。商用製品で使う前にライセンス条件を確認してください。
- ktransformers は今もメンテナンスされていますか?
- 活発にメンテナンス中。ktransformers の最終コミットは 2026年10月20日 です。
- ktransformers は何に使えますか?
- 大容量メモリのCPUサーバーと組み合わせた単一GPUで大規模MoEモデルを実行する CPU-GPU混在のエキスパート配置で、SGLang経由でMoEモデルをサービングする LLaMA-Factoryを使い、少数のコンシューマー向けGPUで超大規模MoEモデルをファインチューニングする
- ktransformers のインストール方法は?
- kt-kernel のソースディレクトリに移動する: cd kt-kernel ソースから kt-kernel をインストールする: pip install .
- ktransformers はどこで入手できますか?
- ソースコードとインストール手順は GitHub にあります:https://github.com/kvcache-ai/ktransformers
概要はプロジェクトの README をもとに AI が生成しています。GitHub の数値は毎日更新されます。
関連プロジェクト
量子化なしで、レイヤー単位に大規模言語モデルを実行し、70Bクラスのモデルを4GB GPU 1枚で推論できるPythonライブラリ。
SGLang、vLLM、TokenSpeedでのLLMサービングを高速化する、投機的デコード用ハーネスEAGLE-4の推論コード。
ルーティングされたエキスパートをディスクから読み込み、一般的なハードウェアで超大規模MoEモデルを動かす純C製推論エンジン。
TurboMindとPyTorchの推論エンジンを備え、LLMや視覚言語モデルの圧縮、デプロイ、サービングを行うツールキット。
DeepSeek-V4.1-Flash-1017(EXL3 2.9 bpw)を2台のNVIDIA DGX Sparksで配信し、1Mトークンのコンテキストに対応したOpenAI互換APIを提供するレシピ。