Would there be interest in having a PyTorch port of https://github.com/materialsvirtuallab/m3gnet in this repo?