20개 이상의 오픈 웨이트 모델을 사전 학습, 파인튜닝, 평가, 서빙할 수 있는 PyTorch 기반 LLM 구현 툴킷입니다.
kvcache-ai/ktransformers
제한된 GPU 메모리에서 대규모 MoE 모델을 CPU-GPU 이기종 환경으로 추론하고 LoRA/전체 파인튜닝하기 위한 연구 프레임워크입니다.
- 스타
- 19,621
- 포크
- 1,602
- 열린 이슈
- 526
- 마지막 커밋
- 그저께
어떤 프로젝트인가요?
KTransformers는 CPU와 GPU에 작업을 분산해 대규모 언어 모델을 실행하고 파인튜닝하는 연구 프로젝트입니다. 추론 기능은 INT4/INT8 양자화 가중치를 위한 CPU 최적화 커널을 제공하며, AMX와 AVX512/AVX2 가속 및 Mixture-of-Experts 모델을 위한 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 모델을 CPU-GPU 이기종 환경으로 추론하고 LoRA/전체 파인튜닝하기 위한 연구 프레임워크입니다. KTransformers는 CPU와 GPU에 작업을 분산해 대규모 언어 모델을 실행하고 파인튜닝하는 연구 프로젝트입니다. 추론 기능은 INT4/INT8 양자화 가중치를 위한 CPU 최적화 커널을 제공하며, AMX와 AVX512/AVX2 가속 및 Mixture-of-Experts 모델을 위한 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에서 추론할 수 있는 Python 라이브러리입니다.
SGLang, vLLM, TokenSpeed에서 LLM 서빙 속도를 높이는 speculative decoding 하네스 EAGLE-4의 추론 코드입니다.
라우팅된 전문가를 디스크에서 스트리밍해 일반 하드웨어에서 초대형 MoE 모델을 실행하는 순수 C 추론 엔진입니다.
TurboMind 및 PyTorch 추론 엔진을 제공하며 LLM과 비전-언어 모델을 압축, 배포, 서빙하는 툴킷입니다.
DeepSeek-V4.1-Flash-1017(EXL3 2.9 bpw)을 NVIDIA DGX Sparks 두 대에서 서빙하는 레시피로, 1M 토큰 컨텍스트를 지원하는 OpenAI 호환 API를 제공합니다.