TensorFlow Framefwork API mirrors the overall structure of Python Keras.
Please check
https://github.com/tensorflow/java/blob/master/tensorflow-framework/README.md