https://medium.com/@monocosmo77/updates-on-sharpness-aware-minimization-part1-machine-learning-2023-438f095ec177