https://medium.com/@monocosmo77/updates-on-sharpness-aware-minimization-part3-machine-learning-2023-663bea8565b7