Open-sourcing the code associated with the NeurIPS 2025 paper "Efficient Transition Matrices to Enable State Tracking in State-Space Models" - View it on GitHub