MATH500 Train => MATH500_train.jsonl from https://github.com/openai/prm800k/blob/main/prm800k/math_splits/train.jsonl
MATH500 Test => MATH500_test.jsonl from https://github.com/openai/prm800k/blob/main/prm800k/math_splits/test.jsonl