nuwa/stage: links for crossing-flash-attn