torch>=1.10.0
torchvision>=0.11.1
numpy
pyyaml>=5.4.1
scipy
cython
pycocotools>=2.0.2
torchdistill>=0.2.7
compressai>=1.1.8
timm>=0.4.12

[test]
pytest
