Accelerating machine learning with training data management