Add neural network training scheduler