diskcache>=5.6.1
pandas>=1.0.0
scikit-learn>=1.0.0

[all]
bpemb>=0.3.3
diskcache>=5.6.1
gensim>=4.3.1
keras-nlp>=0.6.0
openai>=0.25.0
pandas>=1.0.0
scikit-learn>=1.0.0
sense2vec==2.0.0
sentence-transformers>=2.2.2
timm>=0.6.7

[bpemb]
bpemb>=0.3.3
diskcache>=5.6.1
pandas>=1.0.0
scikit-learn>=1.0.0

[cohere]
cohere>=4.11.2
diskcache>=5.6.1
pandas>=1.0.0
scikit-learn>=1.0.0

[dev]
black>=19.3b0
bpemb>=0.3.3
datasets==2.8.0
diskcache>=5.6.1
flake8>=3.6.0
gensim>=4.3.1
interrogate>=1.5.0
keras-nlp>=0.6.0
matplotlib==3.4.3
mkdocs-material==9.1.21
mkdocs==1.5.2
mkdocstrings-python==1.3.0
mkdocstrings==0.22.0
mktestdocs==0.1.2
mktestdocs==0.1.2
openai>=0.25.0
pandas>=1.0.0
pre-commit>=2.2.0
pytest-xdist
pytest>=4.0.2
scikit-learn>=1.0.0
sense2vec==2.0.0
sentence-transformers>=2.2.2
timm>=0.6.7

[gensim]
diskcache>=5.6.1
gensim>=4.3.1
pandas>=1.0.0
scikit-learn>=1.0.0

[keras_nlp]
diskcache>=5.6.1
keras-nlp>=0.6.0
pandas>=1.0.0
scikit-learn>=1.0.0

[openai]
diskcache>=5.6.1
openai>=0.25.0
pandas>=1.0.0
scikit-learn>=1.0.0

[pytorch]
diskcache>=5.6.1
pandas>=1.0.0
scikit-learn>=1.0.0
torch>=1.12.0

[sense2vec]
diskcache>=5.6.1
pandas>=1.0.0
scikit-learn>=1.0.0
sense2vec==2.0.0

[sentence-tfm]
diskcache>=5.6.1
pandas>=1.0.0
scikit-learn>=1.0.0
sentence-transformers>=2.2.2

[spacy]
diskcache>=5.6.1
pandas>=1.0.0
scikit-learn>=1.0.0
spacy>=3.5.0

[text]
bpemb>=0.3.3
diskcache>=5.6.1
gensim>=4.3.1
keras-nlp>=0.6.0
pandas>=1.0.0
scikit-learn>=1.0.0
sense2vec==2.0.0
sentence-transformers>=2.2.2

[vision]
diskcache>=5.6.1
pandas>=1.0.0
scikit-learn>=1.0.0
timm>=0.6.7
