Add mlflow callback for pushing config to mlflow artifacts (#1125)

Add mlflow callback for pushing config to mlflow artifacts (#1125) #2854