dill replacement and keras fix

-cloudpickle is replaced with dill to allow pickling locked objects. -keras is replaced with tensorflow.keras, as for the latest version (2.5.0), the optimizers can not be accessed from keras (from keras.optimizers import SGD) is invalid. To be consistent, the code uses only tensorflow.keras