Public repository for "The Surprising Effectiveness of Test-Time Training for Abstract Reasoning" - View it on GitHub