Thank you for your contribution.
However, I don't think I can merge theses changes due to the following reasons:
Replacing Keras with Pytorch means that we won't be able to use multiple backends. At least in my configuration, Jax seems to perform faster than Pytorch and Tensorflow, so I think its good to have the choice to use a different backend.
As for the logs, I don't think it makes much adding them to the repository.
Merge de ambas ramas