PrimeIntellect‑ai가 분산 머신러닝 학습에 쓸 수 있는 Dion 및 Muon 옵티마이저의 효율적인 구현을 담은 GitHub 저장소 dion을 공개했다. 이 코드는 최신 PyTorch(v2.7 이상)에서 작동하며 DDP, FSDP2, 텐서 병렬 처리 등 DTensor 기반 병렬화를 지원한다. pip를 이용해 저장소에서 패키지를 바로 설치할 수 있다. README에는 GPU 8개로 GPT-small 모델을 학습하는 예제 스크립트도 포함됐다.
왜 중요한가
연구자들이 바로 사용할 수 있는 통신 효율적인 옵티마이저를 제공해 대규모 모델 학습 속도를 높일 수 있다.