davda54 / sam Python 2K 209 SAM: Sharpness-Aware Minimization (PyTorch) optimizer pytorch sam sharpness-aware