"submatrix" -> "local matrix"? is `size` the number of rows? How does it work with `n_loc_cols`?