nuwa/stage: links for pytorchviz