Generalize mapSpec to contexts #32
Description
Currently mapSpec only supports unary functions. It would be nice if we could make users write a forwardPropagate function instead of mapSpec that's generalized to non-unary functions. That would in turn improve the plan back-propagation.