latest working version with LME using pytorch-lightning 1.8.
Before first nwgraph message passing refactoring (1.0.0).