Code and data for paper "(How) do Language Models Track State?" - View it on GitHub