I read about Ax in a post and I found it to be very powful. But I do have a question to raise. I found the analytical model booth funciton is adopted as evaluation function in the Getting Started section. I wonder if we could use other customized models such as neural networks, GBM etc. with the goal to find the feature vector that can maximize the model output. If we can, is there a way that we directly call these models in the optimize API?

Thanks for your time.