Lightning module clean-up

Consolidate first draft of the lightning module:

• Name is currently MTANLitModule but it can be used to train models that do not include MTan
• forward() abstract method seems useless
• resume_from_checkpoint parameter is not used