Grokking at the edge of linear separability