* The following config keys have been moved in 158 :
* `cnn_keys`, `mlp_keys`, `per_rank_batch_size`, `per_rank_sequence_length`, `per_rank_num_batches` and `total_steps` have been moved to the specifig `algo` config
* We have added the integration of the [MLflowLogger](https://lightning.ai/docs/pytorch/stable/api/lightning.pytorch.loggers.mlflow.html) in #159 . This comes with new documentation and notebooks under the `example` folder on how to use it.