Train TensorFlow models on YARN in just a few lines of code! - View it on GitHub