Generalization at the Edge of Stability