https://quadexcel.com/wp/tips-and-tricks-for-distributed-large-model-training/