We do not have an objective which can be used for solubility prediction - yet. Consider the [MMSol](https://pubs.acs.org/doi/full/10.1021/acs.jcim.5c00748) method which has an open source codebase available with pre-trained model. Check if licensing allows usage.
