folktables~=0.0.12
scikit-learn>=1.3
numpy
pandas
tqdm
accelerate
transformers>=4.42.4
jinja2
torch
sentencepiece
protobuf
netcal
cloudpickle
matplotlib
seaborn

[apis]
litellm>=1.40.14
openai>=1.1.0

[cluster]
htcondor
xgboost

[docs]
sphinx<8,>=6
sphinx_rtd_theme>=2.0
sphinx-autopackagesummary
myst-parser
sphinx-copybutton
sphinxcontrib-bibtex
sphinxemoji
sphinx-autodoc-typehints
numpydoc
nbsphinx
sphinx_gallery
IPython
ipykernel

[tests]
pytest
coverage
flake8-pyproject
flake8
mypy
tox

[vllm]
vllm>=0.6.0
