MegaTrain: 단일 GPU에서 100B+ 파라미터 LLM의 Full Precision 학습
(arxiv.org)
MegaTrain은 CPU 메모리를 활용해 단일 GPU에서도 100B 이상의 파라미터를 가진 LLM을 Full Precision으로 학습할 수 있게 함으로써, GPU 메모리 한계를 극복하고 대규모 모델 개발의 비용 부담을 획기적으로 낮추는 메모리 중심 시스템입니다.
이 글의 핵심 포인트
- 1단일 H200 GPU와 1.5TB 호스트 메모리로 최대 120B 파라미터 모델 학습 가능
- 214B 모델 학습 시 DeepSpeed ZeRO-3 대비 1.84배 높은 처리량 달성
- 3