PyYAML
--index-url https://pypi.python.org/simple/

torch>=1.9
pytorch_lightning>=1.9.0
protobuf==3.20.2
hydra-core>=1.1.0
hydra-colorlog>=1.1.0
numpy
ase>=3.21
torchmetrics
h5py
tqdm
pytest
pytest-datadir
pyrootutils
hydra-optuna-sweeper
optuna
progressbar
sympy
torch_ema
rich
fasteners
pre-commit
black
