@kunduruanil Hi,
As far as I understand, mlflow.autolog() works in basically the same way with Workspace and Unity Catalog experiments. It automatically logs supported parameters, metrics, models, and other information from the ML training run.
The main difference is how the experiments and MLflow assets are managed and governed. Workspace experiments are tied to the workspace, while Unity Catalog gives you more centralized governance and access control.
For production use, Unity Catalog can be a better option when multiple teams need controlled access and governance.
I would also check the Databricks Runtime and MLflow versions, since some features and configurations may depend on the version you are using.
Hope this helps!