nuwa/stage: links for jax-lightning