"At some point AI may be better at math than humans in all ways."
That would be like saying that a calculator is better at maths than humans. It is true in a sense, but we shouldn't personify AI, it has no drive and motivation. The best results in future will come from mathematicians that are adept at using AI to help them, and in optimising the network design. That includes optimising results per KWh.