Improving Formal Reasoning of Transformer with State Stack | Read Paper on Bytez