Megatron-LMは2Bから462Bパラメータのモデルを数千GPU規模で学習でき、H100クラスタで最大47%のMFUを達成しています。Colossal-AIやHugging Face Accelerate等、多くのLLM開発フレームワークがMegatron-LMの設計に影響を受けています。
大規模モデル学習フレームワークを比較したい相談
業界の他フレームワーク設計への影響力の大きさ