GitHub repository: sayakpaul/Sharpness-Aware-Minimization-TensorFlow
Primary language: Jupyter Notebook
Description: Implements sharpness-aware minimization (https://arxiv.org/abs/2010.01412) in TensorFlow 2.