Pipeline Parallelism with Controllable Memory Penghui Qi, Min Lin