torchmd
schnetpack