torch>=2.1.0
pytorch-lightning>=2.1.0
smact>=3.1.0
torch-geometric
torchmetrics
streamlit
dash
ase
tqdm
wandb
click
fire
transformers
pydantic
einops
jupyterlab
pymatgen
