Hyperparameter tuning via trajectory predictions: Stochastic prox-linear methods in matrix sensing