aenum>=2.2.0
diskcache>=5.4.0
lark>=1.0.0
numpy>=1.19.5
pandas>=1.1.5
psutil
pydantic<2
pyyaml>=5.1
retry>=0.9.2
sqlalchemy-utils>=0.36.6
sqlalchemy>=2.0.0
thefuzz

[chromadb]
chromadb

[dev]
Pillow>=8.4.0
PyDriller
PyGithub
black>=23.1.0
boto3
bs4
codespell
coveralls>=3.0.1
eva-decord>=0.6.1
facenet-pytorch>=2.5.2
faiss-cpu
faiss-cpu
flake8>=3.9.1
flaml[automl]
gpt4all
ipython<8.13.0
ipywidgets>=7.7.2
isort>=5.10.1
kornia
langchain
matplotlib>=3.3.4
mock
moto[s3]>=4.1.1
nbmake>=1.2.1
nest-asyncio>=1.5.6
neuralforecast
norfair>=2.2.0
openai>=1.0
opencv-python-headless>=4.6.0.66
pdfminer.six
protobuf
pylint
pymupdf<1.23.0
pytest-benchmark
pytest-cov>=2.11.1
pytest-testmon
pytest>=6.1.2
pytube
replicate
scikit-learn
semantic_version
sentence-transformers
sentencepiece
sentencepiece
statsforecast
timm>=0.6.13
torch>=1.10.0
torchvision>=0.11.1
transformers
transformers
twine
ultralytics>=8.0.93
wheel>=0.37.1
youtube-transcript-api

[document]
bs4
faiss-cpu
gpt4all
langchain
openai>=1.0
pdfminer.six
protobuf
pymupdf<1.23.0
sentence-transformers
sentencepiece
transformers

[forecasting]
neuralforecast
statsforecast

[function]
boto3
facenet-pytorch>=2.5.2
kornia
norfair>=2.2.0
pytube
youtube-transcript-api

[ludwig]
ludwig[distributed,hyperopt]

[milvus]
pymilvus>=2.3.0

[notebook]
ipython<8.13.0
ipywidgets>=7.7.2
matplotlib>=3.3.4
nbmake>=1.2.1
nest-asyncio>=1.5.6

[pinecone]
pinecone-client

[postgres]
psycopg2

[qdrant]
qdrant_client

[ray]
ray<2.5.0,>=1.13.0

[sklearn]
scikit-learn

[vision]
Pillow>=8.4.0
eva-decord>=0.6.1
faiss-cpu
opencv-python-headless>=4.6.0.66
sentencepiece
timm>=0.6.13
torch>=1.10.0
torchvision>=0.11.1
transformers
ultralytics>=8.0.93

[xgboost]
flaml[automl]
