torch>=1.1.0
packaging
setuptools
dependencies==2.0.1
typing-extensions>=3.7.4

[:python_version < "3.7"]
dataclasses

[finn_integration]
bitstring
onnxruntime==1.4.0
finn-base
toposort

[finn_integration:python_version < "3.8" or sys_platform == "linux"]
onnx==1.5.0

[finn_integration:python_version >= "3.8" and sys_platform != "linux"]
onnx

[hadamard]
scipy

[pyxir_integration]
onnx

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

[test]
pytest
pytest_cases
pytest-mock
mock
hypothesis
scipy
torchvision
onnxruntime
onnx

[tts]
soundfile
librosa
numpy
scipy
tqdm
pillow
pyyaml

[vision]
torchvision
