Update lightning backend

[SBouchard01]

pytorch_lighning was renamed lightning. This might mess up calls in some envs, we should update the package name in the code files and make sure the methods we use are still up to date.

Note

The pytorch_lighning install still works at the moment, I'm using this one in the install pipeline but some more recent parts of the pipeline call lightning and this could eventually be the cause of discrepancies