nuwa/stage: links for torch-agdiff