nuwa/stage: links for argmax