The currently time limiting factor for diff_predictor is the amount of time hyperparameter optimization takes, especially when not using a GPU, which might limit users who dont have GPU access or experience. We could use Optuna: https://optuna.org/#code_examples to make this process faster