# infra
pytest==7.4.2
pytest-cov==4.1.0
comet_ml==3.35.3
tqdm==4.66.1
rich==13.6.0
pre-commit==3.5.0
ruff==0.2.2

# core
h5py==3.10.0
numpy==1.24.2
torch==2.2.1
lightning==2.2.0
jsonargparse[signatures,argcomplete]==4.32.0
torchmetrics==1.2.1
onnx==1.16.0
onnxruntime==1.15.1
atlas-ftag-tools==0.2.4
scipy==1.12.0
lion-pytorch==0.2.2
