nuwa/stage: links for mlrose-torch