Revert D24524219: Remove `balance` and `devices` parameter from Pipe.
Test Plan: revert-hammer
Differential Revision:
D24524219 (https://github.com/pytorch/pytorch/commit/8da75763032fb74f16c92fd9a8c98e69cd0b9ce3)
Original commit changeset: 9973172c2bb7
fbshipit-source-id: b187c80270adb2a412e3882863a2d7de2a52ed56
