torch>=1.12.0
transformers>=4.20.0
pytorch-lightning>=1.7.0
wandb>=0.13.0
pandas>=1.4.0
numpy>=1.21.0
pymatgen>=2022.5.0
scikit-learn>=1.1.0 