There have been several proposals for improved and simplified high-level APIs for building ... the primary TensorFlow API as of TensorFlow 2.0. The comments in the code above are worth reading.