pandas>=2.0.1
numpy>=2.1
scikit-learn>=0.21.2
torch>=2.5.1
torchvision>=0.20.1
packaging
pydantic>=2.5.0
torchmetrics>=1.1.2

[dev]
pytest>=4.3.0
pytest-markdown-docs
black==24.3.0
isort==5.12.0
pre-commit>=1.15.2
mypy>=0.971
pylint>=2.13.9
flake8
pandas-stubs
types-PyYAML
types-Pillow
types-pywin32
types-requests
types-setuptools
types-tabulate
types-tqdm

[lightning]
lightning>=2.0.8
wandb>=0.15.11

[metrics]
openai-clip

[models]
braindecode
x_transformers>=1.27.20
dalle2_pytorch
vit_pytorch
