대규모 모델의 분산 학습과 추론을 더 쉽고 효율적으로 지원하는 딥러닝 최적화 라이브러리입니다.
ServiceNow/Fast-LLM
PyTorch와 Triton 기반의 오픈소스 라이브러리로, 3D 병렬 처리와 ZeRO 메모리 최적화, Hugging Face 연동을 지원해 대규모 언어 모델을 학습합니다.
- 스타
- 337
- 포크
- 53
- 열린 이슈
- 56
- 마지막 커밋
- 6일 전
어떤 프로젝트인가요?
Fast-LLM은 PyTorch와 Triton을 기반으로 대규모 언어 모델을 학습하는 오픈소스 라이브러리입니다. 약 10억 개의 파라미터를 가진 모델부터 700억 개 이상의 파라미터를 가진 모델을 학습하는 클러스터까지 지원하며, 단일 GPU부터 Slurm 또는 KubeFlow를 사용하는 Kubernetes로 관리되는 멀티 노드 클러스터까지 실행할 수 있습니다.
이 라이브러리는 데이터·텐서·파이프라인 병렬 처리, 시퀀스 길이 병렬 처리, 메모리 효율을 위한 ZeRO-1·ZeRO-2·ZeRO-3를 지원합니다. 모델은 YAML 파일로 설정하고 CLI에서 실행합니다. 또한 토큰을 버리지 않는 Mixture-of-Experts 구현이 포함되어 있으며 Hugging Face Transformers와 연동됩니다.
프로젝트는 Apache 2.0 라이선스를 따르며, 빠른 배포를 위해 사전 빌드된 Docker 이미지를 제공합니다. 문서와 예제에서는 README에 사용된 Mistral-7B 학습 설정을 다룹니다.
핵심 기능
- 01데이터·텐서·파이프라인(3D) 병렬 처리를 통한 분산 학습.
- 02ZeRO-1·ZeRO-2·ZeRO-3 메모리 최적화.
- 03토큰을 버리지 않는 Mixture-of-Experts(MoE) 구현.
- 04혼합 정밀도 학습 및 그래디언트 누적.
- 05CLI를 통한 YAML 기반 설정.
- 06Hugging Face Transformers 연동.
활용 사례
- Slurm으로 관리되는 멀티 노드 GPU 클러스터에서 Mistral 계열 또는 GPT 스타일 모델 사전 학습.
- KubeFlow PyTorchJob을 사용해 Kubernetes에서 대규모 LLM 학습 작업 실행.
- Apache 2.0 라이선스의 오픈소스 코드베이스로 사용자 정의 MoE 모델 학습.
빠른 시작
- 1
nvcr.io/nvidia/pytorch:25.11-py3 Docker 이미지를 모든 노드에 배포합니다(권장)
- 2
pip으로 모든 노드에 Fast-LLM을 설치합니다
pip install --no-cache-dir "fast-llm[CORE,OPTIONAL] @ git+https://github.com/ServiceNow/Fast-LLM.git" - 3
예제 Slurm 작업 스크립트를 클러스터에 제출합니다
sbatch examples/fast-llm.sbat
사용 전 알아둘 점
- 표시된 규모로 학습하려면 노드마다 A100 또는 H100 GPU 8개를 갖춘 4노드 구성과 같은 멀티노드 GPU 클러스터가 필요합니다.
- 모든 노드에 CUDA 12.1 이상과 PyTorch, Triton, Apex가 설치되어 있어야 합니다.
- README에 따르면 사용자 지정 가능한 옵티마이저와 튜토리얼 등의 일부 기능은 아직 개발 중입니다.
토픽
자주 묻는 질문
- Fast-LLM은(는) 무엇인가요?
- PyTorch와 Triton 기반의 오픈소스 라이브러리로, 3D 병렬 처리와 ZeRO 메모리 최적화, Hugging Face 연동을 지원해 대규모 언어 모델을 학습합니다. Fast-LLM은 PyTorch와 Triton을 기반으로 대규모 언어 모델을 학습하는 오픈소스 라이브러리입니다. 약 10억 개의 파라미터를 가진 모델부터 700억 개 이상의 파라미터를 가진 모델을 학습하는 클러스터까지 지원하며, 단일 GPU부터 Slurm 또는 KubeFlow를 사용하는 Kubernetes로 관리되는 멀티 노드 클러스터까지 실행할 수 있습니다.
- Fast-LLM은(는) 무료로 사용할 수 있나요? 라이선스는 무엇인가요?
- Fast-LLM은(는) Other 라이선스의 오픈소스입니다. 상업용 제품에 사용하기 전에 라이선스 조건을 확인하세요.
- Fast-LLM은(는) 아직 유지보수되고 있나요?
- 활발히 유지보수 중. Fast-LLM의 마지막 커밋은 2026년 10월 15일입니다.
- Fast-LLM은(는) 어디에 사용할 수 있나요?
- Slurm으로 관리되는 멀티 노드 GPU 클러스터에서 Mistral 계열 또는 GPT 스타일 모델 사전 학습. KubeFlow PyTorchJob을 사용해 Kubernetes에서 대규모 LLM 학습 작업 실행. Apache 2.0 라이선스의 오픈소스 코드베이스로 사용자 정의 MoE 모델 학습.
- Fast-LLM는 어떻게 설치하나요?
- nvcr.io/nvidia/pytorch:25.11-py3 Docker 이미지를 모든 노드에 배포합니다(권장) pip으로 모든 노드에 Fast-LLM을 설치합니다: pip install --no-cache-dir "fast-llm[CORE,OPTIONAL] @ git+https://github.com/ServiceNow/Fast-LLM.git" 예제 Slurm 작업 스크립트를 클러스터에 제출합니다: sbatch examples/fast-llm.sbat
- Fast-LLM은(는) 어디서 받을 수 있나요?
- 소스 코드와 설치 방법은 GitHub에 있습니다: https://github.com/ServiceNow/Fast-LLM
요약은 프로젝트 README를 바탕으로 AI가 작성했습니다. GitHub 통계는 매일 갱신됩니다.
관련 프로젝트
대규모 AI 모델의 학습과 추론을 위한 PyTorch 기반 분산 시스템으로, 병렬화, 메모리 최적화, 이기종 학습을 지원합니다.
컴퓨터 비전 작업을 위한 이미지 모델 아키텍처, 사전 학습 가중치, 학습·검증·내보내기 스크립트를 제공하는 PyTorch 라이브러리입니다.
이미지 분류와 관련 비전 작업을 위한 Vision Transformer(ViT) 및 다양한 변형 모델의 PyTorch 구현체입니다.
Mamba는 선택적 상태 공간 모델을 기반으로 한 PyTorch 시퀀스 모델링 아키텍처로, 언어를 비롯한 시퀀스 작업에서 Transformer를 대체할 선형 시간 연산 방식을 제공합니다.
GPT 트랜스포머처럼 병렬 학습도 가능한 RNN 스타일 언어 모델 아키텍처 RWKV의 공식 학습 및 레퍼런스 코드입니다.