https://neurotechnus.com/en/deepspeed-tutorial-scalable-transformers/