Overall reviews of the paper are positive. However, please do take the reviewers’ feedback into account. In particular, adding more discussion about the scalability of the proposed approach. It would also be useful to have a good discussion about how the proposed approach can be combined with FlexFlow to yield both pipeline and parallelism in other dimensions.