numpy>=1.18.2
onnx>=1.7.0
python-dateutil
torch
wrapt
ruamel.yaml
scikit-learn
sentencepiece<1.0.0
tqdm>=4.41.0
numba
wget
frozendict
unidecode

[all]
black==19.10b0
isort[requirements]<5
parameterized
pytest
pytest-runner
ruamel.yaml
sphinx
sphinxcontrib-bibtex
wrapt
wget
wandb
inflect
regex
matplotlib
pypinyin
attrdict
pystoi
pesq
pandas
inflect
pytorch-lightning>=1.5.0
torchmetrics>=0.4.1rc0
transformers>=4.0.1
webdataset<=0.1.62,>=0.1.48
omegaconf>=2.1.0
hydra-core>=1.1.0
pyyaml<6
braceexpand
editdistance
inflect
kaldi-io
librosa
marshmallow
packaging
ruamel.yaml
soundfile
sox
kaldi-python-io
kaldiio
scipy
pandas
g2p_en
pydub
pyannote.core
pyannote.metrics
torch-stft
ipywidgets
matplotlib
pillow
torchvision
boto3
h5py
matplotlib>=3.3.2
sentencepiece
youtokentome>=1.0.5
numpy
rapidfuzz
gdown
inflect
sacrebleu[ja]
sacremoses>=0.0.43
nltk>=3.6.5
wordninja==2.0.0
fasttext
opencc
pangu
jieba
librosa
nltk

[asr]
braceexpand
editdistance
inflect
kaldi-io
librosa
marshmallow
packaging
ruamel.yaml
soundfile
sox
kaldi-python-io
kaldiio
scipy
pandas
g2p_en
pydub
pyannote.core
pyannote.metrics
torch-stft
ipywidgets
matplotlib
pytorch-lightning>=1.5.0
torchmetrics>=0.4.1rc0
transformers>=4.0.1
webdataset<=0.1.62,>=0.1.48
omegaconf>=2.1.0
hydra-core>=1.1.0
pyyaml<6

[core]
pytorch-lightning>=1.5.0
torchmetrics>=0.4.1rc0
transformers>=4.0.1
webdataset<=0.1.62,>=0.1.48
omegaconf>=2.1.0
hydra-core>=1.1.0
pyyaml<6

[cv]
pillow
torchvision
pytorch-lightning>=1.5.0
torchmetrics>=0.4.1rc0
transformers>=4.0.1
webdataset<=0.1.62,>=0.1.48
omegaconf>=2.1.0
hydra-core>=1.1.0
pyyaml<6

[nlp]
boto3
h5py
matplotlib>=3.3.2
sentencepiece
youtokentome>=1.0.5
numpy
rapidfuzz
gdown
inflect
sacrebleu[ja]
sacremoses>=0.0.43
nltk>=3.6.5
wordninja==2.0.0
fasttext
opencc
pangu
jieba
pytorch-lightning>=1.5.0
torchmetrics>=0.4.1rc0
transformers>=4.0.1
webdataset<=0.1.62,>=0.1.48
omegaconf>=2.1.0
hydra-core>=1.1.0
pyyaml<6

[test]
librosa
nltk
pytorch-lightning>=1.5.0
torchmetrics>=0.4.1rc0
transformers>=4.0.1
webdataset<=0.1.62,>=0.1.48
omegaconf>=2.1.0
hydra-core>=1.1.0
pyyaml<6

[text_processing]
inflect
regex

[torch_tts]
matplotlib
pypinyin
attrdict
pystoi
pesq
pandas
inflect

[tts]
librosa
nltk
pytorch-lightning>=1.5.0
torchmetrics>=0.4.1rc0
transformers>=4.0.1
webdataset<=0.1.62,>=0.1.48
omegaconf>=2.1.0
hydra-core>=1.1.0
pyyaml<6
braceexpand
editdistance
inflect
kaldi-io
librosa
marshmallow
packaging
ruamel.yaml
soundfile
sox
kaldi-python-io
kaldiio
scipy
pandas
g2p_en
pydub
pyannote.core
pyannote.metrics
torch-stft
ipywidgets
matplotlib
pytorch-lightning>=1.5.0
torchmetrics>=0.4.1rc0
transformers>=4.0.1
webdataset<=0.1.62,>=0.1.48
omegaconf>=2.1.0
hydra-core>=1.1.0
pyyaml<6
matplotlib
pypinyin
attrdict
pystoi
pesq
pandas
inflect
