Complete PyTorch Lightning training loop.

[TianyuDu]

• [ ] add a PyTorch-Lightning class wrapper for both CLM and NLM, ideally should be a single wrapper class handling both model classes.
• [ ] add a notebook/tutorial demonstration the use of the PyTorch lightning wrapper.