At the moment we canonicalize only MulExpression with matmul_canon but this is does not take into account scalar multiplication. We should have two different functions to canonicalize multiplications instead of one in here.
We could take inspiration from the DGP canonicalizers: