numpy
sympy
ase>=3.21
h5py
pyyaml
hydra-core>=1.1.0
torch>=1.9
pytorch_lightning>=2.0.0
torchmetrics==1.0.1
hydra-colorlog>=1.1.0
rich
fasteners
dirsync
torch-ema
matscipy
tensorboard

[test]
pytest
pytest-datadir
pytest-benchmark
