nuwa/stage: links for torch-ai-2