implement axes_scan's broadcast semantics via dce
By using DCE instead of partial evaluation, we unblock JAX_REMAT3=1 and
the deletion of general partial eval in JAX.
(A JAX version skew issue: in the newest (unreleased) JAX, consts are
just leading invars. So we don't switch on len(consts).)
