make EquilibriumOptimizer use a dict to map ops to their nodes rather than running everything on everything.