numpy>=1.26.4
scipy>=1.5.0
torch>=2.0.1
einops>=0.8.2
fast_pytorch_kmeans>=0.2.2
psutil
pydot>=4.0.0

[dev]
black~=23.12
ruff~=0.14.4
mypy~=1.7
pyrefly~=0.53
pytest
pytest-cov
pre-commit
imageio~=2.37.0
scikit-image~=0.24.0
pytest-xdist
hypothesis>=6.120.0
matplotlib>=3.9.2
scikit-learn>=1.0
build
twine
pip-audit
sphinx>=7.0
furo
sphinx-autodoc-typehints
nbsphinx
sphinx-copybutton
myst-parser
ipykernel
scipy-stubs
pdbpp
nbstripout

[simple-einet-parity]
omegaconf>=2.3
pandas>=2.2
mnist1d
icecream>=2.1.3
tqdm>=4.66
torchvision>=0.20

[sklearn]
scikit-learn>=1.0
