Understanding the Evolution of the Neural Tangent Kernel at the Edge of Stability