anvil -> mtenn: add pytorch lighting wrapper

[khuddzu]

In anvil from openadmet-models we have implemented a pytorch lightning wrapper to train a mtenn schnet model.

Can be ported over.