huggingface-hub<2.0,>=1.3.0
numpy>=1.17
packaging>=20.0
pyyaml>=5.1
regex!=2019.12.17
tokenizers<=0.23.0,>=0.22.0
typer-slim
safetensors>=0.4.3
tqdm>=4.27

[accelerate]
accelerate>=1.1.0

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

[audio]
torchaudio
librosa
pyctcdecode>=0.4.0
phonemizer

[benchmark]
optimum-benchmark>=0.3.0

[chat_template]
jinja2>=3.1.0
jmespath>=1.0.1

[codecarbon]
codecarbon>=2.8.1

[deepspeed]
deepspeed>=0.9.3
accelerate>=1.1.0

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

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

[integrations]
kernels<0.11,>=0.10.2
optuna
codecarbon>=2.8.1

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

[kernels]
kernels<0.11,>=0.10.2

[num2words]
num2words

[open-telemetry]
opentelemetry-api
opentelemetry-exporter-otlp
opentelemetry-sdk

[optuna]
optuna

[quality]
datasets>=2.15.0
ruff==0.14.10
GitPython<3.1.19
urllib3<2.0.0
libcst
rich

[retrieval]
faiss-cpu
datasets>=2.15.0

[sagemaker]
sagemaker>=2.31.0

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

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

[sklearn]
scikit-learn

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

[tiktoken]
tiktoken
blobfile

[timm]
timm>=1.0.23

[torch]
torch>=2.4
accelerate>=1.1.0

[video]
av

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