Toshniwal, S.; Wiseman, S.; Livescu, K.; Gimpel, K. Chess As a Testbed for Language Model State Tracking. AAAI 2022, 36, 11385-11393.