sharpness aware minimization

Instead of directly optimizing the loss, optimize for weights that lie in regions of weight space where the loss is uniformly small.