This is a nice package to have. So are there any plans to upgrade it for TF Keras? I would like to contribute if you are planning to do so. Let me know your thoughts.