There is [torch-md](https://github.com/torchmd/torchmd)