2024
Transformers Use Causal World Models in Maze-Solving Tasks.
CoRR, 2024

2023
Structured World Representations in Maze-Solving Transformers.
CoRR, 2023

A Configurable Library for Generating and Manipulating Maze Datasets.
CoRR, 2023

Linearly Structured World Representations in Maze-Solving Transformers.
Proceedings of UniReps: the First Workshop on Unifying Representations in Neural Models, 2023

2022
Sparse Relational Reasoning with Object-Centric Representations.
CoRR, 2022