cattrs~=22.2
colorlog~=6.9
dacite~=1.6
Mako~=1.2
numpy<3,>=1.26
pandas~=2.0
pyhocon~=0.3.59
ubelt~=1.3
retrying~=1.3
spacy~=3.5
tqdm~=4.64
zstandard~=0.18.0
sqlitedict<3.0,>=2.1.0
bottle~=0.12.23
datasets~=3.1
pyarrow>=11.0.0
pyarrow-hotfix~=0.6
nltk!=3.9.0,~=3.7
rouge-score~=0.1.2
scipy>=1.10
uncertainty-calibration~=0.1.4
scikit-learn>=1.1
transformers~=4.53
torch<3.0.0,>=1.13.1
torchvision<3.0.0,>=0.14.1

[accelerate]
accelerate~=0.25

[aleph-alpha]
aleph-alpha-client~=2.14
tokenizers>=0.13.3

[all]
medhelm[proxy-server]
medhelm[scenarios]
medhelm[metrics]
medhelm[plots]
medhelm[decodingtrust]
medhelm[slurm]
medhelm[cleva]
medhelm[images]
medhelm[models]
medhelm[mongo]
medhelm[heim]
medhelm[vlm]
medhelm[codeinsights]
medhelm[lmkt]

[allenai]
ai2-olmo~=0.2

[amazon]
boto3~=1.34
awscli~=1.33
botocore~=1.34

[anthropic]
anthropic~=0.41
websocket-client~=1.3

[audiolm]
medhelm[openai]
medhelm[google]
pydub~=0.25.1
ffmpeg-python~=0.2.0
soundfile~=0.12
librosa~=0.10
einops~=0.7.0
openai-whisper==20240930
transformers_stream_generator~=0.0.4
av~=14.3
scipy~=1.10
torchvision<3.0.0,>=0.14.1
flash-attn~=2.7
pycocoevalcap~=1.2
jiwer~=3.0
rapidfuzz~=3.10
jieba~=0.42.1

[ci]
medhelm[metrics]
medhelm[openai]
medhelm[plots]
medhelm[together]
medhelm[yandex]
medhelm[cohere]
medhelm[proxy-server]

[cleva]
unidecode~=1.3
pypinyin~=0.49.0
jieba~=0.42.1
opencc~=1.1
langdetect~=1.0

[codeinsights]
clang~=20.1
Levenshtein~=0.27

[cohere]
cohere~=5.3

[decodingtrust]
fairlearn~=0.9.0

[dspy]
dspy~=3.0

[gated]
gdown~=5.1

[google]
google-cloud-aiplatform~=1.48
google-genai~=1.48

[heim]
gdown~=5.1
diffusers~=0.34.0
icetk~=0.0.4
medhelm[openai]
einops~=0.7.0
omegaconf~=2.3
pytorch-lightning~=2.0
ftfy~=6.1
Unidecode~=1.3
wandb~=0.16
google-cloud-translate~=3.11
autokeras~=1.0
clip-anytorch~=2.5
google-cloud-storage~=2.9
lpips~=0.1.4
multilingual-clip~=1.0
NudeNet~=2.0
numpy>=1.26
pytorch-fid~=0.3.0
tensorflow~=2.11
timm~=0.6.12
torch-fidelity~=0.3.0
torchmetrics~=0.11.1
scikit-image==0.*,>=0.22
medhelm[images]

[heim:python_version < "3.10"]
jax~=0.4.30
jaxlib~=0.4.30
flax~=0.8.5
opencv-python-headless<=4.11.0.86,>=4.7.0.68

[heim:python_version >= "3.10"]
jax~=0.6.2
jaxlib~=0.6.2
flax~=0.10.7
opencv-python<4.8.2.0,>=4.7.0.68

[human-evaluation]
scaleapi~=2.13
surge-api~=1.1

[ibm]
ibm-watsonx-ai~=1.2

[ibm-enterprise-scenarios]
openpyxl~=3.1

[image2struct]
medhelm[images]
latex~=0.7.0
pdf2image~=1.16
selenium~=4.17
html2text~=2024.2.26
opencv-python-headless<=4.11.0.86,>=4.7.0.68
lpips~=0.1.4
imagehash~=4.3

[images]
medhelm[accelerate]
pillow>=10.2

[litellm]
litellm>=1.80.0

[lmkt]
sentence_transformers~=4.1

[medhelm]
accelerate~=0.25
medhelm[openai]
medhelm[yandex]
medhelm[scenarios]
bert_score~=0.3.13
lxml~=5.3
openpyxl~=3.1
python-docx~=1.1

[metrics]
google-api-python-client~=2.64
numba~=0.56
sacrebleu~=2.2
langdetect~=1.0
immutabledict~=4.2
gradio_client~=1.3

[mistral]
mistralai~=1.1

[models]
medhelm[ai21]
medhelm[accelerate]
medhelm[aleph-alpha]
medhelm[allenai]
medhelm[amazon]
medhelm[anthropic]
medhelm[cohere]
medhelm[google]
medhelm[mistral]
medhelm[openai]
medhelm[reka]
medhelm[together]
medhelm[yandex]
medhelm[writer]

[mongo]
pymongo~=4.2

[openai]
openai~=2.8
tiktoken~=0.7
pydantic~=2.0

[plots]
colorcet~=3.0
matplotlib>=3.6.0
seaborn>=0.11.0

[proxy-server]
gunicorn>=20.1

[ranking]
pytrec_eval==0.5

[reka]
reka-api~=2.0

[scenarios]
gdown~=5.1
xlrd~=2.0

[seahelm]
pythainlp==5.0.0
pyonmttok==1.37.0
sacrebleu~=2.2
python-crfsuite~=0.9.11

[slurm]
simple-slurm~=0.2.6

[summarization]
summ-eval~=0.892
bert-score~=0.3
rouge-score~=0.1.2
nltk!=3.9.0,~=3.7
sentencepiece~=0.2.0
protobuf

[together]
together~=1.1

[unitxt]
evaluate~=0.4.1

[vlm]
medhelm[openai]
einops~=0.7.0
einops-exts~=0.0.4
open-clip-torch~=2.24
torch~=2.1
transformers_stream_generator~=0.0.4
scipy~=1.10
torchvision<3.0.0,>=0.14.1
medhelm[reka]
medhelm[images]
medhelm[image2struct]
pycocoevalcap~=1.2
qwen-vl-utils~=0.0.8

[writer]
writerai~=4.0

[yandex]
sentencepiece~=0.2.0
