karpathy/nanochat
Python1 / 0
단일 GPU 노드에서 토큰화부터 채팅 UI까지 LLM의 전 과정을 실험할 수 있는 최소형 학습 프레임워크이다.
핵심 포인트
- depth 파라미터 하나로 모델 크기에 따른 최적의 하이퍼파라미터를 자동 설정한다.
- 8개의 H100 GPU 노드에서 약 2시간 만에 GPT-2급 성능인 DCLM CORE 점수 0.2565 이상을 달성한다.
- COMPUTE_DTYPE 설정을 통해 torch.amp.autocast 없이 하드웨어별 최적 정밀도를 명시적으로 제어한다.
- 사전 학습된 모델에 정체성을 부여하는 합성 데이터 생성 및 SFT와 RL 과정을 통합 제공한다.
48k
STARS
6.3k
FORKS
+928
TRENDING
1
조회수
관련 토론
아직 관련 토론이 없습니다.
댓글
댓글을 작성하려면 로그인이 필요합니다.