numpy>=1.18.0
scipy>=1.4.1
scikit-learn<1.1.0,>=0.22.2
six
setuptools
tqdm
numba>=0.53.1

[all]
mxnet
catboost
lightgbm
tensorflow
tensorflow-addons
h5py
torch
torchvision
xgboost
pandas
kornia
matplotlib
Pillow
statsmodels
pydub
resampy
ffmpeg-python
cma
librosa
opencv-python

[catboost]
catboost

[docs]
sphinx>=1.4
sphinx_rtd_theme
sphinx-autodoc-annotation
sphinx-autodoc-typehints
matplotlib
numpy>=1.18.0
scipy>=1.4.1
six>=1.13.0
scikit-learn<1.1.0,>=0.22.2
Pillow>=6.0.0

[gpy]
GPy

[keras]
keras
h5py

[lightgbm]
lightgbm

[lingvo_asr]
tensorflow-gpu==2.1.0
lingvo==0.6.4
pydub
resampy
librosa

[mxnet]
mxnet

[non_framework]
matplotlib
Pillow
statsmodels
pydub
resampy
ffmpeg-python
cma
pandas
librosa
opencv-python
pytest
pytest-flake8
pytest-mock
pytest-cov
codecov
requests
sortedcontainers

[pytorch]
torch
torchvision

[pytorch_audio]
torch
torchvision
torchaudio
pydub
resampy
librosa

[pytorch_image]
torch
torchvision
kornia
Pillow
ffmpeg-python
opencv-python

[tensorflow]
tensorflow
tensorflow_addons
h5py

[tensorflow_audio]
tensorflow
tensorflow_addons
h5py
pydub
resampy
librosa

[tensorflow_image]
tensorflow
tensorflow_addons
h5py
Pillow
ffmpeg-python
opencv-python

[xgboost]
xgboost
