Thanks for your answer! At first I though you were using machine learning as an end-to-end approach for solving an optimization task. Now, I understand that you use ML for creating a surrogate model and GD for optimizing it. Interesting!