torch>=2.0.0
torchvision>=0.15.1
numpy
pyyaml>=6.0.0
scipy
cython
pycocotools>=2.0.2
torchdistill>=1.0.0
compressai>=1.2.3
timm>=1.0.3

[docs]
sphinx
sphinx_rtd_theme
sphinxcontrib-youtube

[test]
pytest
