Functional RL with Keras and Tensorflow Eager