matplotlib>=3.9.0
numpy>=2.2.0
onnx>=1.18.0
onnxscript~=0.4.0
Pillow>=11.0.0
polars>=1.31.0
pyarrow>=20.0.0
pycocotools~=2.0.10
rich>=14.0.0
scikit-learn>=1.6.0
scipy>=1.15.0
tensorboard>=2.19.0
torchinfo~=1.8.0
torchmetrics~=1.8.1
tqdm>=4.67.0
webdataset>=0.2.111
torch>=2.7.0
torchvision

[dev]
altair~=5.5.0
bandit~=1.8.6
black~=25.1.0
build~=1.3.0
bumpver~=2025.1131
captum~=0.7.0
coverage~=7.10.5
debugpy
flake8-pep585~=0.1.7
flake8~=7.3.0
invoke~=2.2.0
ipython
isort~=6.0.1
Jinja2~=3.1.5
mkdocs~=1.6.1
mkdocs-exclude~=1.0.2
MonkeyType~=23.3.0
mypy~=1.17.1
parameterized~=0.9.0
pyarrow~=21.0.0
pylint~=3.3.8
requests~=2.32.5
safetensors~=0.6.2
setuptools
torchprofile==0.0.4
twine~=6.1.0
types-requests~=2.32.0
unidecode
urllib3~=2.5.0
wheel

[hf]
huggingface_hub
transformers
