dependencies==2.0.1
packaging
setuptools<70.0
sympy
torch>=1.12
typing-extensions>=3.7.4
unfoldNd

[dev]
pre-commit

[diffusion]
accelerate
diffusers<=0.34
numexpr<2.14.0
pandas
torch>=2.2
torchmetrics[image]
tqdm
transformers<4.57.0

[docs]
m2r2==0.3.3.post2
nbsphinx==0.9.3
nbsphinx-link==1.3.0
pydata-sphinx-theme==0.15.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]
onnxoptimizer
onnxscript
qonnx

[export:python_version < "3.13"]
onnx==1.17.0

[export:python_version >= "3.13"]
onnx

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

[finn_integration:python_version < "3.13"]
onnx==1.17.0

[finn_integration:python_version >= "3.13"]
onnx

[hadamard]
scipy

[lighteval]
lighteval[math]==0.13

[llm]
accelerate
datasets
gguf==0.18.0
numexpr<2.14.0
onnx
onnxruntime<1.21
optimum
pandas
pydantic
torch>=2.4
tqdm
transformers[sentencepiece]<5.0,>=4.52

[notebook]
jupyter
nbmake

[numpy]

[numpy:python_version < "3.13"]
numpy<2.0

[numpy:python_version >= "3.13"]
numpy

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

[ort_integration:python_version < "3.13"]
onnx==1.17.0

[ort_integration:python_version >= "3.13"]
onnx

[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<8.4
pytest-mock
pytest-xdist
pytest_cases
scipy
torchvision
tqdm

[tts]
librosa
numpy
pillow
pyyaml
scipy
soundfile
tqdm

[vision]
accelerate
torch>=1.13
torchvision
tqdm
