torch>=2.0.0
numpy>=1.22
pytorch-lightning>=1.9
tqdm
torchmetrics>=0.11.0

[dev]
black
flake8
importlib_metadata

[test]
coverage
