numpy>=1.19.5
torch>=2.6.0
lightning>=1.4.0
torchmetrics>=1.4.0
tqdm>=4.0.0
traker>=0.3.2
annoy>=1.17.0
datasets>=4.8.4
torcheval>=0.0.6
torchvision>=0.10.0
kronfluence>=1.0.1
transformers>=4.44.0
PyYAML>=5.0.0
captum<0.9,>=0.8.0
Pillow>=10.0.0
pyarrow>=18.0.0
dattri>=0.3.0
tiktoken>=0.10.0

[dev]
coverage>=7.2.3
pytest<=7.4.4
pytest-cov>=4.0.0
pytest-mock==3.10.0
pre-commit>=3.2.0
mypy==1.16.1
types-tqdm>=4.0.0
furo
sphinx-rtd-theme>=1.0.0
sphinx>=7.0.0
numpydoc
tensorboard
wandb>=0.12.10
torchvision>=0.10.0
types-Pillow>=8.3.0
ruff>=0.7.0
types-PyYAML>=5.0.0
hydra-core>=1.1.0
hydra-optuna-sweeper>=1.2.0
hydra-joblib-launcher>=1.2.0
optuna<3.0.0,>=2.10.0

[tutorials]
torchvision>=0.10.0
matplotlib>=3.4.0
pillow>=8.3.0
python-dotenv>=1.0.0
nltk>=3.6.0
wandb>=0.12.10
