Locating the local minima in lens design with machine learning