Replace mlflow and azure-mlflow by custom code.

Replace mlflow and azure-mlflow by custom code. #11811