Multi-stage model predictive control for slug flow crystallizers using uncertainty-aware surrogate models