Scalable multi-node training with TensorFlow Amazon Web Services