NVIDIA Reshard, RL 가중치 전송 최대 5.7배 단축
Reshard: NVIDIA가 공개한, 강화 학습 가중치 전송을 최대 5.7배 가속하는 집단 통신 연산
·2026.11.10 15:30
핵심 내용
NVIDIA가 공개한 Reshard 연산이 분리 배치 RL 환경의 가중치 전송 시간을 최대 5.7배 단축했다.
자세히 보기
- NVIDIA가 2026년 11월 6일 기술 블로그를 통해 집단 통신 연산 Reshard를 공개함
- 학습 GPU와 추론 GPU 간 가중치 전송 시 불필요한 데이터를 줄여 최대 5.7배 성능 향상 달성
- DeepSeek v3 671B 모델 기준 DGX H100에서 14.9초에서 2.6초로 단축
- PyTorch DTensor의 Device Mesh와 Placement 정보를 활용해 겹치는 샤드 영역만 계산하여 전송
- NeMo RL 적용 시 Megatron 학습 백엔드 필수, vLLM 또는 Megatron 생성 백엔드 지원
- BF16 및 특정 FP8/MXFP8 조합 지원, 양자화 가중치는 BF16으로 변환 후 전송
- 공식 문서가 nightly 버전이므로 실제 도입 시 NeMo RL 버전 지원 조건 재확인 필요
이 한국어 요약은 AI가 자동으로 만들었습니다. 원문의 주장과 맥락은 원문에서 확인해 주세요. 저작권은 원저작자에게 있습니다.