Add mlflow callback for pushing config to mlflow artifacts

Add mlflow callback for pushing config to mlflow artifacts #36