Pipeline Parallelism with Controllable Memory. - Penghui Qi; Xinyi Wan; Nyamdavaa Amar; Min Lin (NeurIPS, 2024)