Megatron-LMは、NVIDIAが開発する大規模Transformerモデルの学習用フレームワークです。カスタマイズ可能なネイティブPyTorch学習ループを提供し、数十億〜数千億パラメータ規模のモデル学習に対応します。
大規模言語モデルの学習基盤を知りたい相談
PyTorchネイティブな学習ループという少ない抽象化レイヤー