cython~=3.2
numpy~=2.4
tqdm~=4.67
pyyaml~=6.0

[build]
build~=1.4
auditwheel~=6.6
twine~=6.2

[data]
pydtnn[data-arch]
pydtnn[data-image]
pydtnn[data-lang]

[data-arch]
rapidgzip~=0.16

[data-image]
pillow~=12.2
scipy~=1.17

[data-lang]
spacy~=3.8

[dev]
pydtnn[profile]
pydtnn[format]
pydtnn[test]
pydtnn[lint]
pydtnn[build]

[fhe]
polyhe~=1.0

[format]
autopep8~=2.3
absolufy-imports~=0.3
isort~=8.0
ruff~=0.15

[gpu]
pydtnn[gpu-cupy]
pydtnn[gpu-pycuda]

[gpu-cuda]
nvidia-cublas-cu12~=12.6
nvidia-cuda-runtime-cu12~=12.6
nvidia-cudnn-cu12~=9.10
nvidia-nccl-cu12~=2.29

[gpu-cupy]
cupy-cuda12x~=14.0

[gpu-pycuda]
pycuda~=2026.1

[lint]
flake8~=7.3
flake8-gl-codeclimate~=0.2
flake8-docstrings~=1.7
flake8-annotations~=3.2
flake8-pyproject~=1.2
flake8-debug~=0.3
mccabe~=0.7
pep8-naming~=0.15
pycodestyle~=2.14
pydocstyle~=6.3
pyflakes~=3.4
radon~=6.0
pyright~=1.1

[mpi]
pydtnn[mpi-mpi4py]
pydtnn[mpi-pympi]

[mpi-mpi4py]
mpi4py~=4.1

[mpi-pympi]
pympi~=1.4

[profile]
psutil~=7.2
memray~=1.19

[test]
pytest~=9.0
pytest-xdist~=3.8
pytest-cov~=7.1

[test-torch]
torch~=2.11
torchvision~=0.26
torchmetrics~=1.9
