nuwa/stage: links for pybatch