joblib>=1.4.2
numpy>=1.26.4
PyYAML>=6.0.1
torch>=2.5.1

[mace]
mace-torch
e3nn

[prompt-toolkit]
prompt-toolkit>=3.0.43

[pyg]
torch_geometric>=2.6.1

[test]
pytest>=7.4

[triton]
triton
