use spaces instead of axes, add flatten option

9 jobs for matrix_product_operator_mf in 15 minutes and 11 seconds (queued for 30 seconds)