Scheduling training and prediction operations