Simplify partial diffs during inference