super-gradients/requirements.txt

torch>=1.9.0
tqdm>=4.57.0
boto3>=1.17.15
jsonschema>=3.2.0
Deprecated>=1.2.11
scipy>=1.6.1
matplotlib>=3.3.4
psutil>=5.8.0
tensorboard>=2.4.1
setuptools>=65.5.1
coverage~=5.3.1
torchvision>=0.10.0
sphinx~=4.0.2
sphinx-rtd-theme
torchmetrics==0.8
hydra-core>=1.2.0
omegaconf
onnxruntime==1.15.0
onnx==1.15.0
pillow>=10.2.0
pip-tools>=6.12.1
einops==0.3.2
protobuf==3.20.3
treelib==1.6.1
termcolor==1.1.0
packaging>=20.4
# not directly required, pinned by Snyk to avoid a vulnerability
wheel>=0.38.0
# not directly required, pinned by Snyk to avoid a vulnerability
pygments>=2.7.4
stringcase>=1.2.0
numpy<=1.23
rapidfuzz
json-tricks==3.16.1
onnxsim>=0.4.3,<1.0
data-gradients~=0.3.1
albumentations~=1.3
# not directly required, pinned by Snyk to avoid a vulnerability
fonttools>=4.43.0
# not directly required, pinned by Snyk to avoid a vulnerability
werkzeug>=2.3.8