torch>=2.0
matplotlib>=3.5

[dev]
pytest>=7.0
pytest-cov>=4.0

[distill]
transformers>=4.30
datasets>=2.0

[docs]
mkdocs-material>=9.5
mkdocstrings[python]>=0.24

[experiments]
torchvision>=0.15
