Maybe you just want `_mexpand` (which does multinomial and mul expansion) from simplify.py rather than a full `expand()`?