PyTorch implementation of SAM optimization on simple functions with saddle points @ ICML 2026 "Stability Analysis of Sharpness-Aware Minimization" - View it on GitHub