torchmd
schnetpack
