Ongoing research training transformer language models at scale, including: BERT & GPT-2 - Update `megatron/utils.py` · argonne-lcf/Megatron-DeepSpeed@93e4a51