Speedup Megatron-LM from 10% to 30% with Zero Bubble Pipeline Parallelism

An interesting playground to understand the scheduler is here https://huggingface.co/spaces/sail/zero-bubble-pipeline-para...