Architecture-Aware Minimization (A$^2$M): How to Find Flat Minima in Neural Architecture Search