GitHub repository: simaki/sam
Description: SAM: Sharpness-Aware Minimization (PyTorch)