Implements sharpness-aware minimization (https://arxiv.org/abs/2010.01412) in TensorFlow 2. - View it on GitHub