> Now the challenge is to mimic the human creating the "right" question with high fidelity so the whole ML system returns appropriate search results.

This was always the hard part. If you can generate the "right" question then you can generate the code, and since there are already many good math libraries for solving undergrad problems that is all you have to do.