h5py
torch
torch_geometric
lightning
tqdm
numpy
ase
warp-lang>=1.10.1
setuptools>=82.0.0
