Is there currently a proposal for how new NNVM will support expanding nodes in the forward pass into sub-nodes? For instance, expanding mean into sum+div. This is akin to how FGradient outputs a list of nodes which compute the gradient.
Currently, I have a patch which registers the ExpandCompute pass, but the implementation is too brittle to be upstreamed as-is. It would be good to have something similar to allow gradients for compositional operators.