Welcome @inigohrey! If you're looking into using the full suite of MLFlow features then perhaps

kedro-mlflow

developed by @Galileo-Galilei will be great for your workflow! You'll find the team in #908346260224872480. And I've also spotted this article by our friends at Neptune about using

optuna

with Kedro: https://neptune.ai/blog/kedro-pipelines-with-optuna-hyperparameter-sweeps. They also use our experiment tracking functionality if you're looking for a lightweight application of that.