Training neural networks efficiently with high-level TensorFlow APIs