iohub==0.1.0
torch>=2.1.2
timm>=0.9.5
tensorboard>=2.13.0
lightning>=2.3.0
monai>=1.3.1
jsonargparse[signatures]>=4.20.1
scikit-image
matplotlib>=3.9.0
numpy<2

[dev]
pytest
pytest-cov
hypothesis
ruff
black
profilehooks
onnxruntime

[metrics]
cellpose>=3.0.10
scikit-learn>=1.1.3
torchmetrics[detection]>=1.3.1
ptflops>=0.7

[visual]
ipykernel
graphviz
torchview
