Shu, T. et al.  AGENT: A Benchmark for Core Psychological Reasoning. Proceedings of the 38th International Conference on Machine Learning (2021).