A PyTorch implementation of Sharpness-Aware Minimization for Efficiently Improving Generalization - View it on GitHub