There is a good chance to migrate from keras to pythorch and use native ORT Train lib instead tensorflow. #679
