https://analyticsindiamag.com/how-googles-gpipe-is-using-pipeline-parallelism-for-training-neural-networks/