Ye, T., Xu, Z., Li, Y., & Allen-Zhu, Z. Physics of Language Models: Part 2.1, Grade-School Math and the Hidden Reasoning Process [Data set]