dependencies==2.0.1
numpy<=1.26.4
packaging
setuptools<70.0
sympy
torch>=1.9.1
typing-extensions>=3.7.4
unfoldNd

[dev]
pre-commit

[docs]
m2r2==0.3.3.post2
nbsphinx==0.9.3
nbsphinx-link==1.3.0
pydata-sphinx-theme==0.14.3
sphinx==5.3.0
sphinx-autodoc-typehints==1.21.8
sphinx-gallery==0.10.1
sphinxcontrib-napoleon==0.7
sphinxemoji==0.2.0
lxml==4.9.3

[export]
onnx==1.17.0
onnxoptimizer

[finn_integration]
bitstring
onnx==1.17.0
onnxoptimizer
onnxruntime<1.21,>=1.15.0
qonnx
toposort

[hadamard]
scipy

[llm]
accelerate
datasets
lm_eval
onnx
onnxruntime<1.21
optimum
pandas
tqdm
transformers[sentencepiece]==4.47.1

[notebook]
jupyter
nbmake

[ort_integration]
onnx==1.17.0
onnxoptimizer
onnxruntime<1.21,>=1.15.0
qonnx

[stt]
inflect==5.6.2
librosa
numba
pillow>=4.3.0
requests
ruamel.yaml
soundfile
sox
torch-stft
tqdm
unidecode

[test]
hypothesis
mock
psutil
pytest
pytest-mock
pytest-xdist
pytest_cases
scipy
torchvision
tqdm

[tts]
librosa
numpy
pillow
pyyaml
scipy
soundfile
tqdm

[vision]
accelerate
torchvision
tqdm
