piggyback changes related to GNN flipped taggers to Yinmiao's fork+branch

This MR adds changes to how flipped taggers for GNNs are scheduled to Yinmiao's branch.

Note: the branch is rebased to 23.0

let's see what happens if this gets merged 😄