What about tf.keras? https://www.tensorflow.org/versions/r1.9/programmers_guide/k...