
[accelerate]
accelerate>=1.1.0

[all]
Pillow<=15.0,>=10.0.1
accelerate>=1.1.0
av
blobfile
jinja2>=3.1.0
jmespath>=1.0.1
kernels<0.13,>=0.12.0
librosa
mistral-common[image]>=1.10.0
num2words
phonemizer
protobuf
pyctcdecode>=0.4.0
sentencepiece!=0.1.92,>=0.1.91
tiktoken
timm>=1.0.23
torch>=2.4
torchaudio
torchvision

[audio]
librosa
phonemizer
pyctcdecode>=0.4.0
torchaudio

[benchmark]
optimum-benchmark>=0.3.0

[chat_template]
jinja2>=3.1.0
jmespath>=1.0.1

[codecarbon]
codecarbon>=2.8.1

[deepspeed]
accelerate>=1.1.0
deepspeed>=0.9.3

[deepspeed-testing]
GitPython<3.1.19
accelerate>=1.1.0
beautifulsoup4
datasets>=2.15.0
deepspeed>=0.9.3
dill<0.3.5
evaluate>=0.4.6
faiss-cpu
fastapi
filelock
hf-doc-builder
libcst
mistral-common[image]>=1.10.0
nltk<=3.8.1
openai>=1.98.0
optuna
parameterized>=0.9
protobuf
psutil
pydantic>=2
pytest-asyncio>=1.2.0
pytest-env
pytest-order
pytest-random-order
pytest-rerunfailures<16.0
pytest-rich
pytest-timeout
pytest-xdist
pytest<9.0.0,>=7.2.0
rich
rjieba
rouge-score!=0.0.7,!=0.0.8,!=0.1,!=0.1.1
ruff==0.14.10
sacrebleu<2.0.0,>=1.4.12
sacremoses
sentencepiece!=0.1.92,>=0.1.91
starlette
tensorboard
timeout-decorator
tomli
torch>=2.4
transformers-mlinter==0.1.1
ty==0.0.20
urllib3<2.0.0
uvicorn

[dev]
GitPython<3.1.19
Pillow<=15.0,>=10.0.1
accelerate>=1.1.0
av
beautifulsoup4
blobfile
datasets>=2.15.0
dill<0.3.5
evaluate>=0.4.6
faiss-cpu
fastapi
filelock
fugashi>=1.0
hf-doc-builder
ipadic<2.0,>=1.0.0
jinja2>=3.1.0
jmespath>=1.0.1
kernels<0.13,>=0.12.0
libcst
librosa
mistral-common[image]>=1.10.0
nltk<=3.8.1
num2words
openai>=1.98.0
parameterized>=0.9
phonemizer
protobuf
psutil
pyctcdecode>=0.4.0
pydantic>=2
pytest-asyncio>=1.2.0
pytest-env
pytest-order
pytest-random-order
pytest-rerunfailures<16.0
pytest-rich
pytest-timeout
pytest-xdist
pytest<9.0.0,>=7.2.0
rhoknp<1.3.1,>=1.1.0
rich
rjieba
rouge-score!=0.0.7,!=0.0.8,!=0.1,!=0.1.1
ruff==0.14.10
sacrebleu<2.0.0,>=1.4.12
sacremoses
scikit-learn
sentencepiece!=0.1.92,>=0.1.91
starlette
tensorboard
tiktoken
timeout-decorator
timm>=1.0.23
tomli
torch>=2.4
torchaudio
torchvision
transformers-mlinter==0.1.1
ty==0.0.20
unidic>=1.0.2
unidic_lite>=1.0.7
urllib3<2.0.0
uvicorn

[docs]
hf-doc-builder

[integrations]
codecarbon>=2.8.1
kernels<0.13,>=0.12.0
optuna

[ja]
fugashi>=1.0
ipadic<2.0,>=1.0.0
rhoknp<1.3.1,>=1.1.0
unidic>=1.0.2
unidic_lite>=1.0.7

[kernels]
kernels<0.13,>=0.12.0

[mistral-common]
mistral-common[image]>=1.10.0

[num2words]
num2words

[optuna]
optuna

[quality]
GitPython<3.1.19
datasets>=2.15.0
libcst
rich
ruff==0.14.10
tomli
transformers-mlinter==0.1.1
ty==0.0.20
urllib3<2.0.0

[retrieval]
datasets>=2.15.0
faiss-cpu

[sagemaker]
sagemaker>=2.31.0

[sentencepiece]
protobuf
sentencepiece!=0.1.92,>=0.1.91

[serving]
accelerate>=1.1.0
fastapi
openai>=1.98.0
pydantic>=2
rich
starlette
torch>=2.4
uvicorn

[sklearn]
scikit-learn

[testing]
GitPython<3.1.19
accelerate>=1.1.0
beautifulsoup4
datasets>=2.15.0
dill<0.3.5
evaluate>=0.4.6
faiss-cpu
fastapi
filelock
hf-doc-builder
libcst
mistral-common[image]>=1.10.0
nltk<=3.8.1
openai>=1.98.0
parameterized>=0.9
protobuf
psutil
pydantic>=2
pytest-asyncio>=1.2.0
pytest-env
pytest-order
pytest-random-order
pytest-rerunfailures<16.0
pytest-rich
pytest-timeout
pytest-xdist
pytest<9.0.0,>=7.2.0
rich
rjieba
rouge-score!=0.0.7,!=0.0.8,!=0.1,!=0.1.1
ruff==0.14.10
sacrebleu<2.0.0,>=1.4.12
sacremoses
sentencepiece!=0.1.92,>=0.1.91
starlette
tensorboard
timeout-decorator
tomli
torch>=2.4
transformers-mlinter==0.1.1
ty==0.0.20
urllib3<2.0.0
uvicorn

[tiktoken]
blobfile
tiktoken

[timm]
timm>=1.0.23

[torch]
accelerate>=1.1.0
torch>=2.4

[video]
av

[vision]
Pillow<=15.0,>=10.0.1
torchvision
