Can we use AxSearch with tensorflow models?

I notice that Ax internally applies BoTorch that is for Bayesian optimization in Pytorch. If my model is written in Tensorflow, can I use AxSearch to search hyperparameters? Will there be some conflicts?

That should work fine!

yes, I just finished a multi-day Ax search on a Keras model.

1 Like