This PR adds support for chained matmuls. In
order to do this, we implement the following
Support for op-specific vector_shapes
Support for op-specific dim expansions (reduction
dims of mmas will now always have chained expansions,
even if they are not the induction variable of
reduction ops)
This PR adds support for chained matmuls. In order to do this, we implement the following