integrate mlflow to track experiments

[shatz01]

Seems like there exists an mlflow logger for pl: https://pytorch-lightning.readthedocs.io/en/stable/api/pytorch_lightning.loggers.mlflow.html