Feature/lightning upgrade

[lorenzomammana]

Summary

Upgrade lightning to version 2.1.*

Refactor hydra plugins to use optional dev groups intend of extras to avoid dragging local packages around in external installations.
Hardcode specific pytorch versions to reduce the search space.
Improve trainer configs to avoid wrong overrides when calling different trainer overrides, add all trainer related configs inside the trainer configurations.
Refactor all lightning related code to support new code design from version 2

Type of Change

• New feature (non-breaking change that adds functionality)
• Breaking change (fix or feature that would cause existing functionality to not work as expected)

Checklist

Please confirm that the following tasks have been completed:

• I have tested my changes locally and they work as expected. (Please describe the tests you performed.)
• I have added unit tests for my changes, or updated existing tests if necessary.
• I have updated the documentation, if applicable.
• I have installed pre-commit and run locally for my code changes.