setuptools<82,>=65.0.0
fastapi[all]
SQLAlchemy
streaming_form_data
alembic
kink
numpy
pandas<3.0.0
joblib
pydantic
pydantic-settings
starlette
scikit-learn<1.8.0
datasets
diffusers
evaluate
accelerate
Pillow
beartype
plotly
shap>=0.48
pymc-bart
arviz<1.0.0
typer
rich
torch
torchvision
transformers
controlnet_aux
sacrebleu
sentencepiece
optuna
cmaes
hyperopt
nvidia-ml-py
openpyxl
httpx
wordcloud
opencv-python
protobuf
imblearn
numba
llvmlite
huey
ijson
greenery==3.2
xlrd
filetype
torchmetrics
pywebview
openml
oslo.concurrency
grad-cam>=1.5.5
dice-ml>=0.12
lime>=0.2.0.1

[cpu]
torch
torchvision

[cpu:sys_platform != "darwin"]
llama-cpp-python

[cpu:sys_platform == "darwin"]
llama-cpp-python==0.3.29

[cuda]
torch
torchvision
llama-cpp-python
