Shared sensitivity to data distribution during learning in humans and transformer networks Paper Code