Comet has an integration with Keras (disclosure: I work for Comet). But it does exactly what you're looking for and it's super easy.

The Comet integration with Keras automatically logs the following items:

• Model and graph description.
• Steps and epochs.
• Metrics (such as loss and accuracy)
• Hyperparameters.
• Optimizer Parameters (such as the learning rate, beta decay rate, and more)
• Number of trainable parameters.
• Histograms for weights and biases.