We note that there seems to be a problem with the processing of the training data in file "LSTM.py". The data "X_train" includes the information of heuristic choice (Is this an intentional design?).
Official code for experiments and (and website) of "Arithmetic Without Algorithms" paper, accepted to ICLR 2025.