Add Suport for mlflow

[AmgadHasan]

⚠️ Please check that this feature request hasn't been suggested before.

• I searched previous Ideas in Discussions didn't find any similar feature requests.
• I searched previous Issues didn't find any similar feature requests.

🔖 Feature description

It would be great to add support to MLFlow for experiment tracking and model versioning.

✔️ Solution

It would be great to add support to MLFlow for experiment tracking and model versioning.

❓ Alternatives

No response

📝 Additional Context

No response

Acknowledgements

• My issue title is concise, descriptive, and in title casing.
• I have searched the existing issues to make sure this feature has not been requested yet.
• I have provided enough information for the maintainers to understand and evaluate this request.

[AmgadHasan]

Can someone point me to where I should add the mlflow support in the code?

[JohanWork]

Hi sorry haven't seen this. I have started adding mlflow see this merged pr and this pr. Feel free to take a look and help out or test it :)

[NanoCode012]

The PR has been merged. Thanks Johan!