https://medium.com/simplified-machine-learning/unlocking-the-power-of-model-parallelism-with-pippy-scaling-ai-to-new-heights-3c7e5fa158a2