SAM: Sharpness-Aware Minimization (PyTorch)
Sharpness-Aware Minimization for Efficiently Improving Generalization