Integrate lightning framework

[mralioo]

Integrate a lightning framework as PyTorch high level abstraction. It's recommended as library in the research community because of flexibility, clarity and most importantly rapid and reproducible ML experimentation.
I will adapt a template from the lightning-hydra-template (it has more then 1.2k stars) The template combines PyTorch Lightning, Hydra and MLflow.

• [x] Adapt and Integrate the template
• [x] Enable MLflow
• [x] Enable Optuna
• [ ] Enable Tensorboard

[mralioo]

Done ! no tensorbaord however , mlflow is enough for tracking