AI studies reveal inner workings of short-term memory