I guess Megatron is a language model framework https://developer.nvidia.com/blog/announcing-megatron-for-tr...