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

[:sys_platform == "linux" and platform_machine != "aarch64"]
triton

[:sys_platform == "win32"]
triton-windows
