sam-pytorch (pypi): NFNets, PyTorch
Keywords: optimizer|pytorch|sharpness|aware|minimization|sam