sparsezoo~=1.6
deepsparse~=1.6
sparseml[torchvision,yolov5]~=1.6
pydantic<2.0.0,>=1.8.2
pyyaml>=5.0.0
click~=8.0.0
tensorboard>=2.0.0
setuptools>=56.0.0
optuna>=3.0.2
onnxruntime-gpu

[dev]
black>=20.8b1
flake8>=3.8.3
isort>=5.7.0
pytest>=6.0.0
wheel>=0.36.2
fastai>=2.7.7

[llm]
llm-foundry==0.2.0
nm-transformers

[nm]
sparsezoo~=1.6
deepsparse~=1.6
sparseml[torchvision,yolov5]~=1.6
