torch-geometric
torch
pymatgen
ase
