Artificial intelligence systems may be good at generating text, recognizing images, and even solving basic math problems—but when it comes to advanced mathematical reasoning, they are hitting a wall.

Working memory is like a mental chalkboard we use to store temporary information while executing other tasks. Scientists worked with more than 200 elementary students to test their working memory, ...