accelerate<2.0,>=1.10.0
aiohttp<3.16,>=3.12
aiofiles<26,>=24.1.0
aioresponses<0.8,>=0.7
backoff<2.3,>=2.2.1
click<8.3.0
datasets<5,>=3.2
hdrhistogram<0.11,>=0.10
jsonlines
lm_eval[wandb]<0.5.0,>=0.4
mlflow>=3.1
numpy<2.4,>=1.26
omegaconf==2.4.0.dev4
packaging
pandas<3,>=2.3
peft<0.18,>=0.17
pexpect<4.10,>=4.9
pillow<11.4,>=11.3
protobuf>=6.32
pycares<5.0.0
pydantic<2.13,>=2.11
responses<0.26,>=0.25
safetensors<0.8,>=0.6
skypilot<0.12,>=0.10.2
tensorboard<2.21,>=2.20
torch<2.9.0,>=2.6
torchao<0.15,>=0.12
torchvision<0.24,>=0.21
tqdm
transformers<4.58,>=4.57
trl<0.27,>=0.24
typer
typing_extensions
uvicorn<0.36.0
wandb<0.24,>=0.21

[aws]
skypilot[aws]

[azure]
skypilot[azure]

[bitnet]
onebitllms>=0.0.3

[ci_cpu]
oumi[dev,docs,gcp,synthesis,torchdata,tune]
vllm<0.11,>=0.10
llama-cpp-python<0.4,>=0.3

[ci_gpu]
oumi[deepspeed,dev,docs,gcp,gpu,synthesis,torchdata,tune]

[ci_gpu:python_version >= "3.10"]
alpaca-eval<0.7,>=0.6

[deepspeed]
deepspeed<0.19.0,>=0.17.0

[dev]
ipykernel
ipywidgets
jupyterlab
nbconvert
nbformat
pre-commit
pyright
pytest-asyncio
pytest-cov
pytest-timeout
pytest
responses
ruff
torchfix

[docs]
myst_parser
myst-nb
sphinx-book-theme
sphinx-copybutton
sphinx-design
sphinx
sphinxcontrib-bibtex
sphinxcontrib-mermaid
sphinxcontrib-typer

[evaluation]
langdetect
immutabledict
nltk>=3.9
antlr4-python3-runtime==4.13.2
sympy>=1.14
sentencepiece>=0.2.0

[evaluation:python_version >= "3.10"]
alpaca-eval<0.7,>=0.6

[file_formats]
pdf2image<1.18,>=1.17
python-poppler<0.5,>=0.4

[flash_attn]
flash-attn>=2.0.0

[gcp]
google-api-core>=2.25
google-api-python-client
google-auth>=2.40
google-cloud-core>=2.4
google-cloud-storage>=3.3
skypilot[gcp]

[gpu]
liger-kernel<0.7,>=0.6
nvidia-ml-py<13.591,>=13.580
bitsandbytes<0.49,>=0.47
vllm<0.11,>=0.10

[gpu:python_version >= "3.10"]
verl<0.6,>=0.5

[lambda]
skypilot[lambda]

[llama_cpp]
llama-cpp-python<0.4,>=0.3.5

[quantization]
autoawq<0.3,>=0.2
bitsandbytes<0.49,>=0.47

[runpod]
skypilot[runpod]

[synthesis]

[synthesis:python_version >= "3.10"]
pdftext>=0.6.3

[torchdata]
torchdata<0.10.0,>=0.9

[tune]
optuna<5.0,>=4.0.0
