Pipeline Parallelism with Controllable Memory - Qi, Penghui; Wan, Xinyi; Amar, Nyamdavaa; Lin, Min (Advances in Neural Information Processing Systems, 2024)