MLflow code example to integrate easily

[cargecla1]

Description

I was trying to use Neuralforecast with MLflow but there wasn't a guide on how to do that. Thus, it will be goo to add an example notebook.

I have a solution to the MLflow integration via additional code on you notebook/ .py . Could anyone from the team reach me out to let me know where I can add an MLflow notebook example with MLflow integration? I have examples with NBEATS, and NHiTS doing autologging with MLflow code added. Please reach out.

Use case

To easily track models, save hyperparameters used for each run and other metadata (e.g. dataset used for the run experiment), compare results and stage models.